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

1. 문제정보

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

프로그래머스

 

2. 풀이

우선 각 숫자를 문자열 형태로 바꾼 후 첫번째 인덱스의 값을 비교하여 정렬을 진행하려고 했다. 하지만 값이 같은 경우 이후 숫자에 대해서도 비교가 필요했고 자릿수나 0과 같은 예외 상황이 있어서 구현하는데 쉽지 않겠다는 생각이 들었다. 예를 들어 [3, 30, 31]이 있는 경우 값을 구하기 위해서는 [31, 3, 30] 으로 정렬이 필요했다. 결국 문제를 해결하지 못했고 다른 분들의 풀이를 참고하여 핵심이 사전순 정렬이라는 것을 알았다.

 

3. 소스코드

from functools import cmp_to_key
def compare(s1, s2):
    s1, s2 = str(s1), str(s2)
    if s1+s2 > s2+s1:
        return 1
    elif s1+s2 < s2+s1:
        return -1
    else:
        return 0


def solution(numbers):
    numbers = sorted(numbers, key=cmp_to_key(compare))
    numbers.reverse()
    numbers = list(map(str, numbers))
    answer = ''.join(numbers)
    if answer[0] == "0":
        return "0"
    return answer

 

4. 배운점

수의 크기, 문자열 길이와 같은 일반적인 정렬 순서가 아닌 경우는 사전순 정렬을 고려해볼 수 있다. 그리고 cmp_to_key를 이용하여 정렬 조건을 커스텀할 수 있다는 것을 배웠다.