Algorithm/Programmers

[프로그래머스] 문자열 정렬하기(1) - Java

뭐든 해보기 2023. 4. 5. 20:03

문제) 문자열 my_string이 매개변수로 주어질 때, my_string 안에 있는 숫자만 골라 오름차순 정렬한 리스트를 return 하도록 solution 함수를 작성해보세요.

 

제한사항)

  • 1 ≤ my_string의 길이 ≤ 100
  • my_string에는 숫자가 한 개 이상 포함되어 있습니다.
  • my_string은 영어 소문자 또는 0부터 9까지의 숫자로 이루어져 있습니다.

입출력 예)

my_string result
"hi12392" [1, 2, 2, 3, 9]
"p2o4i8gj2" [2, 2, 4, 8]
"abcde0" [0]

 

입출력 예 설명)

입출력 예 #1

  • "hi12392"에 있는 숫자 1, 2, 3, 9, 2를 오름차순 정렬한 [1, 2, 2, 3, 9]를 return 합니다.

입출력 예 #2

  • "p2o4i8gj2"에 있는 숫자 2, 4, 8, 2를 오름차순 정렬한 [2, 2, 4, 8]을 return 합니다.

입출력 예 #3

  • "abcde0"에 있는 숫자 0을 오름차순 정렬한 [0]을 return 합니다.

풀이)

1. 정규식 사용하여 문자열에서 숫자만 남기기

2. Stream을 이용해서 문자열을 int 배열로 변환하기

3. Arrays sort 메서드를 이용해 정렬

import java.util.Arrays;
import java.util.stream.Stream;
class Solution {
    public int[] solution(String my_string) {
        my_string = my_string.replaceAll("[^0-9]", "");
		int[] answer = Stream.of(String.valueOf(my_string).split("")).mapToInt(Integer::parseInt).toArray();
		Arrays.sort(answer);
        return answer;
    }
}

 

찾아보니까 Arrays stream 메서드에 정렬도 같이 할 수 있도록 가능하더라,,

 

import java.util.Arrays;
import java.util.stream.Stream;
class Solution {
    public int[] solution(String my_string) {
        my_string = my_string.replaceAll("[^0-9]", "");
        return  Arrays.stream(my_string.split("")).sorted().mapToInt(Integer::parseInt).toArray();
    }
}

 

위처럼 int 배열을 선언하지 않고도 더 간결해질 수 있다.