(Python, Java) 프로그래머스 - JadenCase 문자열 만들기

17 Feb 2022 in Algorithm on Programmers, Level2

[문제 링크]

Python 풀이

def solution(s):
    unit = s.split(" ")
    for idx in range(len(unit)):
        unit[idx] = unit[idx][:1].upper() + unit[idx][1:].lower()
    return ' '.join(unit)

Java 풀이

class Solution {
    public String solution(String s) {
        s = s.toLowerCase();
        String[] unit = s.split("");
        boolean flag = true;

        StringBuilder sb = new StringBuilder();
        for (String letter : unit) {

            if (letter.equals(" ")) {
                flag = true;

            } else if (flag) {
                letter = letter.toUpperCase();
                flag = false;
            }

            sb.append(letter);
        }

        return sb.toString();
    }
}