[프로그래머스] Lv1. 이상한 문자 만들기 파이썬

이상한 문자 만들기

문제 설명

문자열 s는 한 개 이상의 단어로 구성되어 있습니다. 각 단어는 하나 이상의 공백문자로 구분되어 있습니다. 각 단어의 짝수번째 알파벳은 대문자로, 홀수번째 알파벳은 소문자로 바꾼 문자열을 리턴하는 함수, solution을 완성하세요.

제한 사항

• 문자열 전체의 짝/홀수 인덱스가 아니라, 단어(공백을 기준)별로 짝/홀수 인덱스를 판단해야합니다.
• 첫 번째 글자는 0번째 인덱스로 보아 짝수번째 알파벳으로 처리해야 합니다.

입출력 예sreturn

"try hello world"

"TrY HeLlO WoRlD"

내 풀이

def solution(s):
    answer = ''
    s = list(s)
    cnt = 0
    for i in range(len(s)):
        if cnt % 2 == 0:
            s[i] = s[i].upper()
        else:
            s[i] = s[i].lower()
        cnt += 1
        if s[i] == " ":
            cnt = 0
    return ''.join(s)