프로그래머스 - JadenCase 문자열 만들기 (Python)

Lv2, 문자열

문제 설명

JadenCase란 모든 단어의 첫 문자가 대문자이고, 그 외의 알파벳은 소문자인 문자열입니다. 단, 첫 문자가 알파벳이 아닐 때에는 이어지는 알파벳은 소문자로 쓰면 됩니다. (첫 번째 입출력 예 참고)
문자열 s가 주어졌을 때, s를 JadenCase로 바꾼 문자열을 리턴하는 함수, solution을 완성해주세요.

제한 조건

• s는 길이 1 이상 200 이하인 문자열입니다.
• s는 알파벳과 숫자, 공백문자(" ")로 이루어져 있습니다.
  • 숫자는 단어의 첫 문자로만 나옵니다.
  • 숫자로만 이루어진 단어는 없습니다.
  • 공백문자가 연속해서 나올 수 있습니다.

입출력 예시

풀이

• 처음엔 공백이 연속해서 온다는 점을 신경쓰지 못해 틀렸다.
• 오답코드

def solution(s):
    answer = ''
    tmp = list(s.split())
    
    for word in tmp:
        trans = ''
        if 'a' <= word[0] <= 'z':
            trans = word[0].upper() + word[1:].lower()
        else:
            trans = word.lower()
        answer += (trans + " ")
    
    k = len(answer)
    return answer[:k-1]

• split(" ")을 통해 모든 공백을 기준으로 문자열을 분리해주고
• 공백이면 바로 append하고 아니면 변환시키는 방식으로 진행했다.

def solution(s):
    answer = []
    tmp = s.split(" ")
    
    for word in tmp:
        if word:
            answer.append(word[0].upper() + word[1:].lower())
        else:
            answer.append(word)
            
    return " ".join(answer)