[Python/Silver3/2149] 암호 해독

|    문제

어떤 문장을 키를 이용하여 다음과 같이 암호화하려 한다. 암호화하기 전의 문장을 평문이라 하며, 암호화 된 문장은 암호문이라고 한다. 키, 평문, 암호문은 모두 영어 대문자로 된 공백 없는 문장이다.

키의 길이를 N이라고 했을 때, 우선 평문을 N 글자씩 잘라서 다음과 같이 나열한다. 예를 들어 평문이 MEETMEBYTHEOLDOAKTREENTH 이고, 키가 BATBOY라고 해 보자.

BATBOY

M	E	E	T	M	E
B	Y	T	H	E	O
L	D	O	A	K	T
R	E	E	N	T	H

제일 윗줄은 이해를 돕기 위해 키를 다시 한 번 쓴 것이다. 이제 이 행렬(배열)을 열(Column) 단위로 정렬을 하는데, 정렬을 하는 키준은 키의 문자로 한다. 즉 BATBOY를 정렬하여 ABBOTY와 같이 정렬하는 것이다. B와 같이 여러 번 나타나는 문자의 경우에는 원래의 행렬에서 더 왼쪽에 있었던 것을 먼저 쓴다. 정렬을 한 행렬은 다음과 같다.

ABBOTY

E	M	T	M	E	E
Y	B	H	E	T	O
D	L	A	K	O	T
E	R	N	T	E	H

B는 두 가지가 있기 때문에 더 왼쪽에 있었던 (B)MBLR이 먼저 나왔다. 이제 이와 같이 정렬한 행렬을 열 번호가 작은 것 먼저, 열 번호가 같다면 행 번호가 작은 것 순으로 나열하면 암호문이 된다. 즉 위와 같은 경우의 암호문은 EYDEMBLRTHANMEKTETOEEOTH 가 된다.

키와 암호문이 주어졌을 때, 이를 이용하여 평문을 구하는 프로그램을 작성하시오.

 

|    입력

첫째 줄에 키가 주어지고, 둘째 줄에 암호문이 주어진다. 키와 암호문은 모두 영어 대문자로만 되어 있으며, 암호문의 길이가 항상 키의 길이의 배수라고 하자. 키의 길이는 10자 이하이며 암호문의 길이는 100자 이하이다.

 

|    출력

첫째 줄에 평문을 출력한다.

 

 

---

정답 코드

import sys
from collections import defaultdict
input = sys.stdin.readline

class Solution:
    def decrypt(self):
        key = input().strip()
        cypher_text = input().strip()
        key_len, cypher_len = len(key), len(cypher_text)
        cypher_grid = defaultdict(list)
        i = 0
        for k in sorted(key):
            cypher_grid[k].append(cypher_text[i:i+cypher_len//key_len])
            i+=cypher_len//key_len
        
        plain_text = ''
        
        for i in range(cypher_len//key_len):
            check = defaultdict(int)
            for k in key:
                plain_text += cypher_grid[k][check[k]][i]
                check[k] += 1
        print(plain_text)
        

if __name__ == "__main__":
    s = Solution()
    s.decrypt()

 

 

|    비고

 

맨 처음 코드를 작성했을 때, IndexError가 발생하였다. 코드를 확인해보면 변수 j를 처리하는 데에 문제가 있었다. 예를 들어, 예제와 같은 BATBOY나 HUMDING을 입력했을 때는 에러없이 올바르게 출력되지만, AAABBB와 같이 2개 이상의 문자가 중복된 key는 변수 j로만 인덱스를 설정하면 오버될 수 밖에 없었다. 위 정답 코드처럼 defaultdict를 활용하여 알파벳별로 인덱스를 늘리는 것이 인덱스 에러를 발생시키지 않는다.

 

import sys
from collections import defaultdict
input = sys.stdin.readline

class Solution:
    def decrypt(self):
        key = input().strip()
        cypher_text = input().strip()
        key_len, cypher_len = len(key), len(cypher_text)
        cypher_grid = defaultdict(list)
        i = 0
        for k in sorted(key):
            cypher_grid[k].append(cypher_text[i:i+cypher_len//key_len])
            i+=cypher_len//key_len
        
        plain_text = ''
        
        for i in range(cypher_len//key_len):
            j = 0
            for k in key:
                if len(cypher_grid[k]) == 1:
                    plain_text += cypher_grid[k][0][i]
                elif len(cypher_grid[k]) > 1:
                    plain_text += cypher_grid[k][j][i]
                    j += 1

        print(plain_text)

if __name__ == "__main__":
    s = Solution()
    s.decrypt()