Python/Python 문제
둘만의 암호
서영환
2024. 6. 18. 17:52
- 문제 설명
- 예를 들어 s = "aukks", skip = "wbqd", index = 5일 때, a에서 5만큼 뒤에 있는 알파벳은 f지만 [b, c, d, e, f]에서 'b'와 'd'는 skip에 포함되므로 세지 않습니다.
- 따라서 'b', 'd'를 제외하고 'a'에서 5만큼 뒤에 있는 알파벳은 [c, e, f, g, h] 순서에 의해 'h'가 됩니다. 나머지 "ukks" 또한 위 규칙대로 바꾸면 "appy"가 되며 결과는 "happy"가 됩니다.
- 두 문자열 s와 skip, 그리고 자연수 index가 매개변수로 주어질 때 위 규칙대로 s를 변환한 결과를 return하도록 solution 함수를 완성해주세요.
- 제한 조건
- 5 ≤ s의 길이 ≤ 50
- 1 ≤ skip의 길이 ≤ 10
- s와 skip은 알파벳 소문자로만 이루어져 있습니다.
- skip에 포함되는 알파벳은 s에 포함되지 않습니다.
- 1 ≤ index ≤ 20
- 입출력 예
- 실패코드
- 반례 : "abcde", "bcd", 2 , "ffffg" 가 아닌 "fgggg" 가 출력됨
def solution(s, skip, index):
answer = ''
for key in s:
check = ord(key) - 97
i = 0
while i < index:
if chr( ( (check+i) % 26) + 97 ) in skip : check += 1
else: i += 1
answer += chr( ( (check+i) % 26) + 97 )
return answer
- 어려웠던 점
- old()를 이용하여 아스키 값을 가지고 index만큼 더해 주는 간단한 문제로 생각 하여 코드를 작성 하여 테스트가 통과되어 제출을 하였으나 1개를 제외하고는 전부 실패
- 이후 무엇이 문제가 하며 코드를 분석하였지만 어떤 부분에서 문제인지 알기 어렸음
- 질문하기 쪽에 반례들이 있어 그것을 가지고 테스트 케이스 추가하기를 이용하여 반례들을 넣어보니 위의 반례 하나가 실패 뜨는 것을 확인
- 해결 과정
- 반례를 통하여 일단 일단 문자 "abcde"가 주어졌을때 index = 2 만큼 뒤로 이동
- 그러니 check 가 a 가 아닌 b라고 먼저 선언이 필요
- 이후 b가 skip에 있는 확인 있으면 다음 자리인 c로 이동
- 이후 반복하여 skip에 없을 때까지 이동
- 이를 while이 아닌 for i in range(index)를 사용하는게 더 좋을거 같아 while문을 제외 하고 for문을 사용
def solution(s, skip, index):
answer = ''
for key in s:
check = ord(key) #check는 97~122 인 숫자
for i in range(index): #index 만큼 반복
#문자 a가 아닌 b로 변경
check = (check - 96) % 26 + 97
while chr(check) in skip :
# 문자 b가 skip에 포함 되지 않을때까지 반복
# check=b가 skip에 포함 되면 다음 check=c로 변경
check = (check - 96) % 26 + 97
answer += chr( check )
return answer