[프로그래머스] JadenCase 문자열 만들기

문제 링크 : JadenCase 문자열 만들기

문제 설명

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

제한 사항

s는 길이 1 이상 200 이하인 문자열입니다.
s는 알파벳과 숫자, 공백문자(" ")로 이루어져 있습니다.

• 숫자는 단어의 첫 문자로만 나옵니다.
• 숫자로만 이루어진 단어는 없습니다.
• 공백문자가 연속해서 나올 수 있습니다.

내 풀이

public String solution(String s) {
        String answer = "";
        int flag = 0;
        for(int i = 0; i < s.length(); i++){
            if(s.charAt(i) == ' '){
                answer += s.charAt(i);
                flag = 0;
            }else if(flag == 0){
                answer += s.substring(i, i+1).toUpperCase();
                flag = 1;
            }else if(flag == 1){
                answer += s.substring(i, i+1).toLowerCase();
            }
        }
        return answer;
    }

flag 변수를 선언해서 공백 문자를 만났을 때나, 문자열의 맨 처음일때는 toUpperCase 함수를 통해 대문자로 만들고 문자열에 추가하고, 공백일 때는 그대로 문자열에 추가, 문자열의 중간일 때는 toLowerCase 함수를 통해 소문자로 만들고 문자열에 추가한다.
그 후 만들어진 JadenCase 문자열을 반환한다.