[C++백준 1855번:암호]

문제링크

https://www.acmicpc.net/problem/1855

1855번: 암호

문제

준표와 세준이는 서로 솔루션을 토론 하면서 다른 사람이 자신들의 솔루션을 듣지 못하게 하도록 서로 메시지를 주고받을 때 메시지를 암호화 하여서 주고받았다. 암호를 만드는 방법은 다음과 같다. 먼저 암호화 할 문자열을 1,1부터 위에서 아래 순서대로 채운다. 그리고 가장 밑의 행을 채운 후에는 오른쪽 열에서 다시 같은 과정을 반복한다.

만약에 "abcdefghijkl" 이라는 문자열을 3개의 열로 암호화 한다고 하자. 그러면 다음과 같이 표를 채울 수 있을 것이다.

a	e	i
b	f	j
c	g	k
d	h	l

그런 후에는 이제 왼쪽-> 오른쪽, 오른쪽-> 왼쪽, 왼쪽->오른쪽 ... 으로 읽으면서 다시 문자열을 만든다. 위의 경우에는 "aeijfbcgklhd" 가 될 것이다.

우리가 할 일은 다음과 같다. 암호화 된 문자열과 몇 개의 열로 암호화를 하였는지 주어져 있을 때 원래의 문자열을 구하는 프로그램을 작성하여라.

입력

열의 개수 K(1≤K≤20)가 주어지고 두 번째 줄에는 암호화 된 문자열(모두 영소문자)이 주어진다. (문자열의 길이는 200 이하이며 K의 배수이다.)

출력

첫 줄에 원래의 문자열을 출력한다..

 

해석

들어오는 문자열은 항상 열의 배수이다.

[문자열의 길이/열의 개수 * 열의 개수] 의 2차원 배열을 만들어준다.

입력받은 문자열을

짝수 행일때는 열이 커지는 방향으로 넣어주고 

홀수 행일때는 열이 작아지는 방향으로 넣어주면 된다.

읽는 방향은 열 방향으로 순서대로 읽어주면 된다.

코드

#include <iostream>
#include <string>
using namespace std;
 
char arr[10][20];
 
void print(int k, string input) {
    int size = input.size();
    int col = k;
    int row = size / k;
    int num = 0;
 
    for (int i = 0; i < row; i++) {
        for (int j = 0; j < col; j++) {
            if (i % 2 == 1) {
                arr[i][(col - 1) - j] = input[num++];
            }
            else {
                arr[i][j] = input[num++];
            }
 
        }
    }
    for (int i = 0; i < col; i++) {
        for (int j = 0; j < row; j++) {
            cout << arr[j][i];
        }
    }
}
 
int main() {
    int k;
    string input;
    cin >> k;
    cin >> input;
    print(k, input);
}

 

Memo

아직도 문제를 제대로 읽지 못하는 것 같다. 문자열의 길이가 K의 배수인 것을 확인하지 못하고 예외 처리를 어떻게 진행할까 하면서 쓸데없는 고민을 했다..

이전에 제출했던 코드는 메모리 초과가 떴다. 커다란 변수는 전역으로 선언해주는 것이 좋은 것 같다.