[프로그래머스] Lv.0 - 인덱스 바꾸기.py

문제

• 문자열 my_string과 정수 num1, num2가 매개변수로 주어질 때, my_string에서 인덱스 num1과 인덱스 num2에 해당하는 문자를 바꾼 문자열을 return 하도록 solution 함수를 완성해보세요.

제한사항

• 1 < my_string의 길이 < 100
• 0 ≤ num1, num2 < my_string의 길이
• my_string은 소문자로 이루어져 있습니다.
• num1 ≠ num2

입출력 예시

• 입출력 예 1
  • "hello"의 1번째 인덱스인 "e"와 2번째 인덱스인 "l"을 바꾸면 "hlelo"입니다.
• 입출력 예 2
  • "I love you"의 3번째 인덱스 "o"와 " "(공백)을 바꾸면 "I l veoyou"입니다.

코드

def solution(my_string, num1, num2):
    my_string = list(my_string)
    my_string[num1], my_string[num2] = my_string[num2], my_string[num1] 
    answer = ''.join(my_string)
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges