[프로그래머스/Java] Lv.0 문자열 잘라서 정렬하기

문제

문자열 myString이 주어집니다. "x"를 기준으로 해당 문자열을 잘라내 배열을 만든 후 사전순으로 정렬한 배열을 return 하는 solution 함수를 완성해 주세요.

단, 빈 문자열은 반환할 배열에 넣지 않습니다.

- 제한사항

• 1 ≤ myString ≤ 100,000
• myString은 알파벳 소문자로 이루어진 문자열입니다.

풀이

- 오답

어쩐지 간단한 문제인데 질문하기 개수가 많더라니..

엄청 간단한 문제라고 생각했는데, 테스트 케이스는 통과하는데 제출하니까 전부 실패라는 거다. 주어진 입출력 예제인 axbxcxdx, dxccxbbbxaaaa와 달리, x가 2개 이상인 문자열을 처리해야 하는구나 생각하게 되었다. 그제서야 보이는 빈 문자열은 반환할 배열에 넣지 않는다는 문장ㅜ

import java.util.Arrays;

class Solution {
    public String[] solution(String myString) {
        String[] answer = myString.split("x");
        Arrays.sort(answer);
        
        return answer;
    }
}

- 정답

우선 원래대로 split()으로 문자열을 나누어 배열을 반환 받은 뒤, ArrayList 객체를 선언한다. 그다음 확장 for문으로 루프를 돌며 isEmpty()로 빈 문자열이 아닌 경우에만 객체에 add()로 넣어준다.

이후에는 toArray()로 객체를 문자열 배열로 복사한 뒤에 Arrays.sort()로 정렬한다.

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public String[] solution(String myString) {
        String[] splitArr = myString.split("x");
        ArrayList<String> answerArr = new ArrayList<>();
        
        for (String str : splitArr) {
            if (!str.isEmpty()) {
                answerArr.add(str);
            }
        }
        
        String[] answer = answerArr.toArray(new String[answerArr.size()]);
        Arrays.sort(answer);
     
        return answer;
    }
}

결과