프로그래머스 | Lv.2 JadenCase 문자열 만들기 (파이썬)

문제 설명

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

제한 조건

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

입출력 예

s	return
"3people unFollowed me"	"3people Unfollowed Me"
"for the last week"	"For The Last Week"

---

※ 공지 - 2022년 1월 14일 제한 조건과 테스트 케이스가 추가되었습니다.

풀이

첫 글자와 공백 바로 뒤 글자가 대문자가 되는 점을 활용하면 된다.

공백이 연속으로 나올 수도 있기 때문에 공백 다음으로 오는 문자를 체크해야 한다.

def solution(s):
    a=list(s) 
    #전부 소문자로 바꾼다.
    b=s.lower() 
    c=list(b) 
    #첫번째만 대문자로 바꾼다.
    c[0]=a[0].upper() 

    for i in range(len(s)-1):
	  #공백이 있다면 그 다음 문자를 대문자로 바꾼다.
	  if a[i]==' ':
        c[i+1]=a[i+1].upper()
    answer=''.join(c)
  
    return answer