[프로그래머스 - LEVEL2] - 가장 큰 수

링크

가장 큰 수

문제설명

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

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

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

제한 사항

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

풀이

처음 문제를 풀 때 사전상 먼저 오는 문자열로 정렬하면 된다고 생각했다. 하지만 여기서 30, 3이 있었기에 303이 오면 안되고 330이 왔어야 했다. 그렇기에

Arrays.sort(str, new Comparator<String>() {
           public int compare(String str1, String str2)
           {
               if((str1+str2).compareTo(str2+str1)<1)
                   return 1;
               
               return -1;    
           }
        });

처음, 이런식으로 정렬을 해주었지만 테스트케이스에서 런타임에러가 떴다.. 아직까지 런타임에러가 왜 뜬지 모르겠다. 이 부분에서 아시는 분 있으면 알려주시면 감사하겠습니다.

그래서 답지를 보고

        Arrays.sort(str, new Comparator<String>() {
           public int compare(String str1, String str2)
           {
               return (str2+str1).compareTo(str1+str2);
           }
        });

이런식으로 바꾸어주었다.
또한, 예외 케이스로 [0,0,0]가 주어질 수도 있기에
str[0].equals("0")이라면 return 0;을 해주면 된다.

Code

import java.util.*;

class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        String[] str=new String[numbers.length];
        
        for(int i=0; i<numbers.length; i++)
            str[i]=Integer.toString(numbers[i]);
        
        Arrays.sort(str, new Comparator<String>() {
           public int compare(String str1, String str2)
           {
               return (str2+str1).compareTo(str1+str2);
           }
        });
        
        if(str[0].equals("0")) return "0";
        
        for(String kk : str)
            answer+=kk;        
        
        return answer;
    }
}

좋은 웹페이지 즐겨찾기