문제풀이/일일연습문제

(보충필요)99클럽 코테 스터디 7일차 TIL + 프로그래머스/2/42746. 가장 큰 수

Mo_bi!e 2024. 5. 26. 23:47

문제 설명

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"

※ 공지 - 2021년 10월 20일 테스트케이스가 추가되었습니다.

 

 

<코드>

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):
    numbers = list(map(str, numbers))
    
    numbers.sort(key=cmp_to_key
                 (lambda a, b : 
                  (1 if a + b < b + a else 
                   -1 if a + b > b + a else  0)))
    
    return str(int(''.join(numbers)))

 

 

<모범사례>

import functools

def comparator(a,b):
    t1 = a+b
    t2 = b+a
    return (int(t1) > int(t2)) - (int(t1) < int(t2)) #  t1이 크다면 1  // t2가 크다면 -1  //  같으면 0

def solution(numbers):
    n = [str(x) for x in numbers]
    n = sorted(n, key=functools.cmp_to_key(comparator),reverse=True)
    answer = str(int(''.join(n)))
    return answer

 

 

 

<배운점>

단순한 정렬로 안될경우 정렬기준을 커스텀할 때 cmp_to_key 를 이용한다 (커스텀 비교 함수)

이 경우 별도의 식으로 만들어주어여함

 

람다 조건문은 3항연산자를 연달아 물고서 하는 방식으로 이해하면편함