알고리즘 공부/정렬(Sort) | 분류

프로그래머스 정렬 Level 2 (가장 큰 수)

kdhoooon 2021. 4. 10. 19:22

문제


0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

 

 

제한 사항

  • numbers의 길이는 1 이상 100,000 이하입니다.
  • numbers의 원소는 0 이상 1,000 이하입니다.
  • 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

 

풀이


순열로 모든 경우의수를 List에 넣고 List를 정렬하여 문제를 풀었을 때, 답은 나왔으나 제출시에 시간 초과가 나왔다.

 

문제를 쉽게생각해본 결과

숫자 배열을 List 에 넣고 Collection.sort를 이용하여 정렬시켜준 뒤 순서대로 answer 에 넣어주면된다.

 

여기서 정렬의 핵심은 아래 코드와 같다.

Collections.sort(numberToString, new Comparator<String>(){
            
	@Override
    public int compare(String s1, String s2){                
    	return (s2+s1).compareTo(s1+s2); 
    }
});

 

여기서 s2+s1 이 s1+s2 보다 크다면 1, 작다면 -1 이 나온다. 

1이 나오면 s2의 위치가 앞으로 오게 되고, -1이 나오면 s1의 위치가 앞으로 오게된다.

 

[3, 30, 34, 5, 9] 을 예로들면

(303).compareTo(330) = -1

[3, 30, 34, 5, 9]

(343).compareTo(334) = 1

[30, 3, 34, 5, 9]

(53).compareTo(35) = 1

[30, 34, 5, 3, 9]

(93).compareTo(39) = 1

[30, 34, 5, 9 ,3]

 

위와 같은 방식으로 정렬을 마치면,

[ 9, 5, 34, 3, 30] 의 순으로 정렬이 된다.

 

 

<전체코드>

import java.util.*;
import java.io.*;

class Solution {

    public String solution(int[] numbers) {
        String answer = "";
        
        List<String> numberToString = new ArrayList<String>();
        
        for(int i = 0 ; i < numbers.length ; i++){
            numberToString.add(Integer.toString(numbers[i]));
        }
                
        Collections.sort(numberToString, new Comparator<String>(){
            
            @Override
            public int compare(String s1, String s2){                
                return (s2+s1).compareTo(s1+s2); 
            }
        });
        
        if(numberToString.get(0).equals("0"))
            return "0";
        
        answer = numberToString.toString().replaceAll("\\[|\\]","").replaceAll(",", "").replaceAll("\\s", "");
        
        
        return answer;
    }

}