Lv2. 가장 큰 수

JinJinJara·2024년 4월 3일
0

알고리즘 문제 풀이

목록 보기
26/27

문제 설명

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

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

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


제한 사항

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


입출력 예

numbersreturn
[ 6, 10, 2 ]6210
[ 3, 30, 34, 5, 9 ]9534330

풀이

from functools import cmp_to_key
def solution(numbers):
    def compare(x, y):
        return int(str(y) + str(x)) - int(str(x) + str(y))
    res = ''.join(map(str, sorted(numbers, key=cmp_to_key(compare))))
    
    return "0" if int(res) == 0 else res

cmp_to_key()

  • 파이썬의 sorted() 함수의 정렬조건을 compare 함수에서 key로 변환하는 데 사용된다.
  • compare 함수는 따로 정의해야 한다.
  • 함수 인자가 x, y 일 때, return 값에 따라 sorting이 바뀐다.
    • 양수 : y-x
    • 음수 혹은 0 : x-y
  • EX) sorted(iterable, key=cmp_to_key(locale.strcoll))

파이썬 공식문서 : functools 모듈 설명 링크

0개의 댓글