본문 바로가기

알고리즘

99클럽 코테 스터디 4일차 TIL

문제 설명



풀이 과정 & 코드

  • 이전에 역시나 풀어본 문제였음
  • 다 지우고 처음부터 다시 풀어봤는데 내 기억이 맞으면 처음 풀었을 때 낸 실수랑 똑같은 실수를 반복함
  • ex) 주어진 예제가 'a b' 이면 'A B'를 return 해야 하는데 나의 경우에는 'A B'이런식으로 return해서 문제를 계속 틀림
  • 처음 주어진 문자열을 list에 담을때 그냥 split()말고 split(' ')으로 공백까지 리스트에 담아줘야 기존 공백들을 유지할 수 있음
  • 30분 잠깐 다른거를 했는데 그걸 제외해도 30분이나 걸림
  • 그래도 이전보다 더 깔끔한 코드를 작성한 것에 의의를 두는중...
def solution(s):
    strings = list(s.split(' '))
    ans = []
    for i in range(len(strings)):
        if strings[i]:
            ans.append(strings[i][0].upper() + strings[i][1:].lower())
        else:
            ans.append(strings[i])

    return ' '.join(ans)