문제 설명
임의의 문자열이 주어졌을 때 문자 "a", "b", "c"를 구분자로 사용해 문자열을 나누고자 합니다.
예를 들어 주어진 문자열이 "baconlettucetomato"라면 나눠진 문자열 목록은 ["onlettu", "etom", "to"] 가 됩니다.
문자열 myStr이 주어졌을 때 위 예시와 같이 "a", "b", "c"를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.
단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 ["EMPTY"]를 return 합니다.
제한사항
- 1 ≤ myStr의 길이 ≤ 1,000,000
입출력 예
| myStr | result |
| "baconlettucetomato" | ["onlettu", "etom", "to"] |
| "abcd" | ["d"] |
| "cabab" | ["EMPTY"] |
import java.util.*;
class Solution {
public String[] solution(String myStr) {
ArrayList<String> arrList = new ArrayList<String>();
StringBuilder sb = new StringBuilder();
for (int i = 0; i < myStr.length(); i++) {
char ch = myStr.charAt(i);
if (ch != 'a' && ch != 'b' && ch != 'c') {
sb.append(ch);
} else {
if (sb.length() > 0) {
arrList.add(sb.toString());
sb.setLength(0);
}
}
}
// 마지막 문자열 추가 검사
if (sb.length() > 0) {
arrList.add(sb.toString());
}
// 결과가 비어있으면 ["EMPTY"] 반환
return arrList.isEmpty() ? new String[] {"EMPTY"} : arrList.toArray(new String[0]);
}
}
뭘 해도 아무런 오류나 실수 없는 코드가 한 번에 나와서 한 번에 만점을 받는 경우가 없구만...
어떻게 하면 바로
import java.util.Arrays;
class Solution {
public String[] solution(String myStr) {
String[] arr = Arrays.stream(myStr.split("[abc]+")).filter(str -> !str.isEmpty()).toArray(String[]::new);
return arr.length == 0 ? new String[] { "EMPTY" } : arr;
}
}
이런 코드를 적을 수 있을까?
