프로그래머스 12951번 JadenCase 문자열 만들기 - Python
문제 설명
프로그래머스 12951번 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” |
코드 구현
Python
def solution(s: str):
sliced_string = s.split(" ")
answer = []
for string in sliced_string:
string_list = list(string)
if len(string_list) != 0:
string_list[0] = string_list[0].upper()
for index in range(1, len(string_list)):
string_list[index] = string_list[index].lower()
answer.append("".join(string_list))
else:
answer.append(string)
return " ".join(answer)
시간 복잡도 - Python
" "으로 나뉜만큼 반복문을 실행하므로 시간 복잡도는O(n)이다.- 맨 앞 글자가 숫자가 아닌 경우,
" "으로 나뉜 문자열의 길이보다 1 작은 만큼 반복문을 실행하므로 시간 복잡도는O(n)이다. - 따라서, 최악의 경우 시간 복잡도는
O(n^2)이다.
개선할 점 - Python
- 플래그를 사용하여 반복문을 줄이면, 시간 복잡도를
O(n)으로 줄일 수 있다.
def solution(s: str):
answer = []
is_new_word = True
for char in s:
if char == " ":
answer.append(" ")
is_new_word = True
else:
if is_new_word:
answer.append(char.upper())
is_new_word = False
else:
answer.append(char.lower())
return "".join(answer)