[Programmers/Python] JadenCase 문자열 만들기

Programmers - JadenCase 문자열 만들기

문제

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

제한 조건

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

입출력 예

풀이

설계

• 간단하게 입력 문자열 s를 .split(" ")을 이용해 리스트로 만든 뒤, 리스트에서 단어를 하나씩 꺼내 첫 번째 요소만 대문자로 바꿔주면 된다.
• 단, 이때 '연속된 공백'인 경우도 있으므로 공백 역시 최종 답안에 추가해 주어야 한다.
• 또한 문자열의 replace() 메서드는 해당 문자열에 대하여 입력된 인자를 '모두' 치환하므로 사용할 수 없다('aaaa' 같은 경우 'Aaaa'가 아니라 'AAAA'가 된다). 따라서 대소문자 변환 시에는 인덱싱 및 슬라이싱을 이용, (word[0].upper() + word[1:].lower())와 같은 형태로 정답 리스트에 추가한다.
• 정답 리스트는 다시 (" ").join 함수를 이용해 하나의 문자열로 반환한다.

코드

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