본문 바로가기

알고리즘/정렬

[KOTLIN] 가장 큰 수

728x90

- https://programmers.co.kr/learn/courses/30/lessons/42746#

 

코딩테스트 연습 - 가장 큰 수

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요. 예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰

programmers.co.kr

문제 설명

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

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

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

제한 사항

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

입출력 예

numbers return
[6, 10, 2] "6210"
[3, 30, 34, 5, 9] "9534330"

풀이

1. 입력된 numbers 배열의 각 원소를 문자열로 저장한다.

2. 저장된 문자열을 첫번째 char 를 내림차순으로 정렬 한다.

3. 2번에서 정렬 시 첫번째 char 가 같을 경우 두 개의 문자열을 서로 합친 후 Int 형으로 casting 하여 값이 큰 순서로 정렬한다.

코드

class Solution {
    fun solution(numbers: IntArray): String {
        var answer = ""
        val list = mutableListOf<Number>()
        
        numbers.forEach { number ->
            list.add(Number(number.toString()))
        }
        
        list.sort()
        
        if (list[0].value == "0") {
            answer = list[0].value
        } else {
            list.forEach { l ->
                answer += l.value
            }    
        }
        
        return answer
    }
}

private class Number(val value: String) : Comparable<Number> {
    override fun compareTo(other: Number): Int {
        return when {
            value[0] - other.value[0] < 0 -> 1
            value[0] - other.value[0] > 0 -> -1
            else -> {
                // 앞자리가 같을 경우 비교되는 두 값의 string 값을 더한 후 int 비교 한다.
                val value1 = value
                val value2 = other.value
                //println("number : $value1, $value2")
                ((value2 + value1).toInt()).compareTo((value1 + value2).toInt())
            }
        }
    }
}
728x90

'알고리즘 > 정렬' 카테고리의 다른 글

[KOTLIN] H-Index  (0) 2022.03.25