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