[COS Pro 2급] 알파벳 바꾸기 / 파이썬

문제 설명

---

문자열이 주어졌을 때, 문자열에서 'a'는 'z'로, 'z'는 'a'로 바꾸려고 합니다. 예를 들어 주어진 문자열이 "abz"라면 "zba"라는 문자열을 만들면 됩니다.

문자열 s가 매개변수로 주어졌을 때, 문자열에서 'a'는 'z'로, 'z'는 'a'로 바꾸어 return 하도록 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다.

주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

매개변수 설명

---

문자열 s가 solution 함수의 매개변수로 주어집니다.

• 문자열 s는 알파벳 소문자로만 이루어져 있으며 길이는 1 이상 100 이하입니다.

return 값 설명

---

solution 함수는 문자열에서 'a'는 'z'로, 'z'는 'a'로 바꾼 후 return 합니다.

 

 

예시 설명

---

예시

s	result
"abz"	"zba"

설명

예시 #1
첫 번째 알파벳 'a'를 'z'로 바꾸고, 세 번째 알파벳 'z'를 'a'로 바꾸면 "zba"가 됩니다.

작성 코드

---

def solution(s):
    s_lst = list(s)
    n = len(s)
    for i in range(n):
        if s_lst[i] == 'a':
            s_lst[i] = 'z'
        elif s_lst[i] == 'z':
            s_lst[i] =  'a'
    return "".join(s_lst)