암호

문제

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

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

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

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

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

입력

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

출력

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

해결방법

• 마음 편하게 예제 입력을 기준으로 행렬을(열은 입력, 행은 문자 길이를 n으로 나눈 몫 ) 만들고(행이 짝수이면 그대로 입력, 홀수이면 반대로 입력)만들었다.
• 그 행을 열 우선 출력했다.

n = int(input())
s = input()
l = len(s)
#예제와 같은 행렬을 만든다. 
lst = [[0]*(n) for _ in range(l//n)]

x = -1
#행의 길이만큼 돌면서 
for i in range(l//n) :
    #짝수이면 그대로 넣고 
    if i %2 ==0 :
        for j in range(n) :
            x+=1 
            lst[i][j] = s[x]
    #홀수이면 반대로 넣는다. 
    else :
        for j in range(n-1,-1,-1) :
            x += 1
            lst[i][j] = s[x]
result = []
#열 우선으로 출력한다. 
for j in range(n) :
    for i in range(l//n) :
        result.append(lst[i][j])

print(''.join(map(str,result)))