프로그래머스 / 이상한 문자 만들기

Question

문제링크

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

제한사항

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

Input & Output

s	return
"try hello world"	"TrY HeLlO WoRlD"

Logic

1. 각 문자에 대해 공란이면 그대로 추가하고 count 초기화 한다.
2. 다음 반복문에서 0이 되기 위해 -1로 초기화한다.
3. 각 단어에 대해 홀수번이면 소문자로, 짝수번이면 대문자로 만든다.

Code

def solution(s):
    answer = ''
    count=0
    for i in s:
        if i==' ':
            answer+=' '
            count=-1
        elif count%2==1:
            answer+=i.lower()
        else :
            answer+=i.upper()
        count+=1
    return answer