[Programmers] Lv.2 JadenCase 문자열 만들기 (Go)

https://school.programmers.co.kr/learn/courses/30/lessons/12951

프로그래머스

 

문제 설명

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

 

제한사항

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

---

Solution

import (
	"strings"
	"unicode"
)

func solution(s string) string {
	var result strings.Builder
	var idx = 0

	for _, st := range s {

		if unicode.IsSpace(st) {
			result.WriteRune(st)
			idx = 0
			continue
		}

		if idx == 0 {
			result.WriteString(strings.ToUpper(string(st)))
		} else {
			result.WriteString(strings.ToLower(string(st)))
		}

		idx++
	}

	return result.String()
}