[프로그래머스] 이상한 문자 만들기 (Python)

🔗 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12930

문제 설명

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

제한 사항

• 문자열 전체의 짝/홀수 인덱스가 아니라, 단어(공백을 기준)별로 짝/홀수 인덱스를 판단해야합니다.

• 첫 번째 글자는 0번째 인덱스로 보아 짝수번째 알파벳으로 처리해야 합니다.

입출력 예

s return

입출력 예 설명

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

✔풀이

def solution(s):
    answer = ''
    
    words = s.split(' ')
    
    for word in words:
        for i in range(len(word)):
            if i % 2 == 0:
                answer += word[i].upper()
            elif i % 2 == 1:
                answer += word[i].lower()
        answer += ' '
    
    answer = answer[:-1]
    return answer

✔개선 코드

def solution(s):

      result = []

      for word in s.split(' '):

          tmp = ''

          for i, ch in enumerate(word):

              tmp += ch.upper() if i % 2 == 0 else ch.lower()

          result.append(tmp)

      return ' '.join(result)

개선 포인트

1. 문자열 += 연결의 비효율

파이썬에서 문자열은 불변(immutable)이기 때문에 +=를 할 때마다 새로운 문자열 객체가 생성된다.

리스트에 모아서 한 번에 join하는 것이 더 효율적

2. enumerate 활용

range(len(word))로 인덱스를 구한 뒤 word[i]로 접근하는 대신, enumerate를 사용하면 인덱스와 문자를 동시에 가져올 수 있다.

 for i, ch in enumerate(word):

      tmp += ch.upper() if i % 2 == 0 else ch.lower()

3. 마지막 공백 제거 불필요

기존에는 반복문 안에서 answer += ' '로 공백을 추가한 뒤, 마지막 공백을 answer[:-1]로 잘라냈다.

' '.join()을 사용하자. return ' '.join(result)