[프로그래머스 : PYTHON] 이상한 문자 만들기

문제 설명

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

제한 사항

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

입출력 예

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

입출력 예 설명

"try hello world"는 세 단어 "try", "hello", "world"로 구성되어 있습니다. 각 단어의 짝수번째 문자를 대문자로, 홀수번째 문자를 소문자로 바꾸면 "TrY", "HeLlO", "WoRlD"입니다. 따라서 "TrY HeLlO WoRlD" 를 리턴합니다.

 

문제풀이

문제가 복잡해보이지만 단어들의 첫 인덱스가 0이라고 생각하면 어렵지 않다.

flag를 만들어서 문자열의 요소가 공백일 때는 flag = 0으로 초기화해주고 문자열을 거치면서 flag를 0과 1로 바꾸어주며 대소문자로 변환해준다.

 

다른 사람 풀이 : 공백을 기준으로 단어를 나눠서 map를 가지고 각 단어에 enumerate로 인덱스를 새롭게 부여해주고 인덱스에 따라 대소문자 변환을 해준다. 단어마다 공백도 필요하므로 " ".join 도 사용한다.

 

전체코드

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

 

다른 사람 풀이

def solution(s):

    return " ".join(map(lambda x: "".join([a.lower() if i % 2 else a.upper() for i, a in enumerate(x)]), s.split(" ")))