프로그래머스 Lv2 | 정렬 | 가장 큰 수

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

프로그래머스

 

 

2026.04.03

처음에 permutation했다가 numbers 크기보고 아차 싶었다..

그래서 어떻게 하지 하다가,, cmt_to_key 함수는 기억 못할 것 같아서 *3을 해버림

def solution(numbers):
    # permutation을 하기에 numbers 배열이 너무 크다
    # 최적화해서 해야 함
    # 누가 크다고 여길까? 첫번째 나오는 숫자가 큰 애가 앞으로 가야 함
    # 근데 주의해야 할 잠, 0이 들어간 경우와 그렇지 않은 숫자가 있을 때, 10 vs 1 이라면 -> 1이 먼저 오는게 맞음
    numbers_str = [str(num) for num in numbers]
    numbers_str.sort(key=lambda x: (x*3), reverse=True)
    answer = ''.join(numbers_str)
    
    return answer

 

더 간단하게 표현도 가능해

-1이면 앞, 1이면 뒤, 0이면 같음을 의미

from functools import cmp_to_key

def solution(numbers):
    # permutation을 하기에 numbers 배열이 너무 크다
    # 최적화해서 해야 함
    # 누가 크다고 여길까? 첫번째 나오는 숫자가 큰 애가 앞으로 가야 함
    # 근데 주의해야 할 잠, 0이 들어간 경우와 그렇지 않은 숫자가 있을 때, 10 vs 1 이라면 -> 1이 먼저 오는게 맞음
    numbers_str = [str(num) for num in numbers]
    numbers_str.sort(key=cmp_to_key(lambda a,b: (b+a>a+b)-(b+a<a+b)))
    answer = ''.join(numbers_str)
    
    return answer

 

 

---

시도 1 >> 시간 초과 (버블정렬) >>> 60

def sort(str_numbers):
    for i in range(len(str_numbers)):
        for j in range(i+1, len(str_numbers)):
            c_1 = str_numbers[i] + str_numbers[j]
            c_2 = str_numbers[j] + str_numbers[i]
            if c_1 < c_2: # SWAP
                str_numbers[i], str_numbers[j] = str_numbers[j], str_numbers[i]
                
def solution(numbers):
    str_numbers = [str(num) for num in numbers]
    sort(str_numbers)
    answer = ''
    for str_n in str_numbers:
        answer+=str_n
    
    return answer

 

시도 2 >> cmp_to_key

cmp_to_key는 파이썬에서 정렬할 때 기존의 비교 함수(comparator)를 키 함수(key function)로 바꿔주는 도구

from functools import cmp_to_key

def compare(a, b):
    if a + b < b + a:
        return 1
    elif a + b > b + a:
        return -1
    else:
        return 0

def solution(numbers):
    str_numbers = [str(num) for num in numbers]
    str_numbers.sort(key=cmp_to_key(compare))
    answer = ''.join(str_numbers)
    return '0' if answer[0] == '0' else answer