제로베이스 20220422 코테 연습

1. 암호

• Vigenere cipher이라는 암호화 방법은 암호화하려는 문장 (평문)의 단어와 암호화 키를 숫자로 바꾼 다음, 평문의 단어에 해당하는 숫자에 암호 키에 해당하는 숫자를 더하는 방식이다.

• 이 방법을 변형하여 평문의 단어에 암호화 키에 해당하는 숫자를 빼서 암호화하는 방식을 생각해 보자.

• 예를 들어 암호화 키가 love이고, 암호화할 문장이 “nice day” 라면 다음과 같이 암호화가 이루어진다.

• 제시된 평문의 첫 번째 문자인 ‘n’은 해당 암호화 키 ‘l’의 알파벳 순서가 12 이므로 알파벳상의 순서에서 ‘n’보다 12앞의 문자인 ‘b’로 변형된다.

• 변형된 문자가 ‘a’ 이전의 문자가 되면 알파벳 상에서 맨 뒤로 순서를 돌린다.
  • 예를 들면 평문의 세 번째 문자인‘c’는 알파벳 상에서 3 번째이고 대응하는 암호화키 ‘v’는 알파벳 순서 22로 ‘c’에서 22 앞으로 당기면 ‘a’보다 훨씬 앞의 문자이어야 하는데, ‘a’앞의 문자가 없으므로 ‘z’로 돌아가 반복되어 ‘g’가 된다.
  • 즉 평문의 문자를 암호화키의 문자가 알파벳 상에서 차지하는 순서만큼 앞으로 뺀 것으로 암호화한다.

• 평문의 문자가 공백 문자인 경우는 그 공백 문자를 그대로 출력한다.

• 이와 같은 암호화를 행하는 프로그램을 작성하시오.

입력

• 첫째 줄에 평문이, 둘째 줄에 암호화 키가 주어진다.

• 평문은 알파벳 소문자와 공백문자(space)로만 구성되며, 암호화 키는 알파벳 소문자만으로 구성된다.
• 평문의 길이는 공백까지 포함해서 30000자 이하이다.

출력

• 첫 번째 줄에 암호문을 출력한다.

풀이

import sys

arr = list(map(ord,sys.stdin.readline().strip('\n')))
key = list(map(ord,sys.stdin.readline().strip('\n')))

alpha = [chr(x) for x in range(97,123)]

str = ''

for i in range(0,len(arr)):
    if i > len(key)-1:
        k = i - len(key) * (i // len(key))
    else:
        k = i

    if arr[i] != 32:
        str += alpha[arr[i]-key[k]-1]
    else:
        str += chr(32)

print(str)

채점 결과

• 정답

리뷰

• 암호문과 키값의 아스키코드값을 구하고, 해당 값의 차이를 구해서 반환해주면 된다.

• 이 문제의 경우, 반례 첫줄에 개행문자가 없는 데이터가 있는지 일반적인 strip()함수를 쓸 경우 오답이 발생한다.

  • strip(‘\n’)을 통해 정확히 개행문자만 제거해주어야 정답이 나온다.