[프로그래머스] 문자열 잘라서 정렬하기 - JAVA

프로그래머스 _ 코딩 기초 트레이닝 DAY 18 - (2)

출처 - https://school.programmers.co.kr/learn/courses/30/lessons/181866

프로그래머스

문제 | 문자열 myString이 주어집니다. "x"를 기준으로 해당 문자열을 잘라내 배열을 만든 후 사전순으로 정렬한 배열을 return 하는 solution 함수를 완성해 주세요. 단, 빈 문자열은 반환할 배열에 넣지 않습니다.

입력 #1 |  "dxccxbbbxaaaa"

출력 #1 | ["aaaa","bbb","cc","d"]

---

>> 풀이 <<

split을 이용해 x를 기준으로 문자열을 자른 후, 오름차순정렬을 해주면 되는 문제이다!

 

• string.split(" ")
  • split 인수에 자르는 기준을 넣어주면 된다! 여러번 정리했으니 생략
• Arrays.sort()
  • Arrays.sort(arr)은 arr 배열을 오름차순으로 정렬해주는 메소드이다. arr 자체를 정렬해주는 것이기 때문에 별도의 배열을 추가하지 않아도 된다.

이렇게 간단하게 풀 수 있을 줄 알았는데 제출만 하면 틀렸다고 나왔다!!!!!

찾아보니까 dxccxbbbxxaaa 라고 주어질 때, split("x")를 하게 되면, ["d","cc","bbb","","aaa"]가 된다. 그래서 저 빈 요소를 빼는 작업까지 진행해야 했다.

 

빈 요소인지 확인하는 작업은 for문과 isEmpty()를 이용해주었다.

---

코드는 다음과 같다.

import java.util.*;
class Solution {
    public String[] solution(String myString) {
        String[] result = myString.split("x");
        Arrays.sort(result);
        
        List<String> lst = new ArrayList<>();
        for (String str : result) {
            if (!str.isEmpty()) {
                lst.add(str);
            }
        }
        return lst.toArray(new String[lst.size()]);
    }
}

원래 stream을 쓰려고 했는데 filter()에다가 str != null 이라고 써서 제대로 작동이 안 되었다. 여기에도 isEmpty를 쓰면 된다!

 

import java.util.*;
class Solution {
    public String[] solution(String myString) {
        return Arrays.stream(myString.split("x")).filter(s -> !s.isEmpty()).sorted().toArray(String[]::new);
    }
}