백준 4246번 To and Fro // C++

문제

 

Mo and Larry have devised a way of encrypting messages. They first decide secretly on the number of columns and write the message (letters only) down the columns, padding with extra random letters so as to make a rectangular array of letters. For example, if the message is “There’s no place like home on a snowy night” and there are five columns, Mo would write down

Note that Mo includes only letters and writes them all in lower case. In this example, Mo used the character ‘x’ to pad the message out to make a rectangle, although he could have used any letter.

Mo then sends the message to Larry by writing the letters in each row, alternating left-to-right and right-to-left. So, the above would be encrypted as

toioynnkpheleaigshareconhtomesnlewx

Your job is to recover for Larry the original message (along with any extra padding letters) from the encrypted one.

입력

There will be multiple input sets. Input for each set will consist of two lines. The first line will contain an integer in the range 2 . . . 20 indicating the number of columns used. The next line is a string of up to 200 lower case letters. The last input set is followed by a line containing a single 0, indicating end of input.

출력

Each input set should generate one line of output, giving the original plaintext message, with no spaces.

 

---

 

풀이

 

숫자와 문자열을 입력받으면, 다음과 같은 순서로 문자를 나열한 뒤, 세로로 한줄씩 출력하면 되는 문제이다.

---

 

코드

#include <iostream>
#include <string>
#include <vector>

using namespace std;

int main() {
	int num;
	string str;
	while (true) {
		cin >> num;

		if (num == 0)
			break;

		vector<char> *v = new vector<char>[num];

		cin >> str;

		int line = 0;
		
		for (int i = 0; i < str.length(); i++) {
			if (i != 0 && i%num == 0)
				line++;
			if(line % 2 == 0)
				v[i % num].push_back(str[i]);
			else {
				v[num - (i % num) - 1].push_back(str[i]);
			}
		}

		for (int i = 0; i < num; i++) {
			for (int j = 0; j < v[i].size(); j++) {
				printf("%c", v[i][j]);
			}
		}
		printf("\n");
	}
}