[프로그래머스] 가장 큰 수 - python[ 정렬 ]

[level 2] 가장 큰 수 - 42746

문제 링크

성능 요약

메모리: 10.2 MB, 시간: 0.04 ms

구분

코딩테스트 연습 > 정렬

채점결과

Empty

문제 설명

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일 테스트케이스가 추가되었습니다.

나의 틀린 풀이

numbers의 원소는 0 이상 1,000 이하입니다. 라는 조건을 보고 모든 숫자를 4자리로 늘려서 정렬해주는 방식을 사용했다.

결과적으로는 1~6번 문제에 대해서 실패가 발생했는데 풀이 로직에 문제가 있었나보다.

반례를 생각해봤는데 1213 + 12 의 경우 내 로직에는 1222 vs 1213이 되는데 이 결과로 121213이 될 것이다.

다만 정답은 121312가 더 크다는 결론이 나오기 때문에 로직 자체가 잘못된 것이다.

def solution(numbers):

    expand = []
    answer = ''
    for number in numbers:
        length = len(str(number))
        expandedNum = str(number)
        while len(expandedNum) <4:
            expandedNum += expandedNum[-1]
        expand.append([number, int(expandedNum)])
    expand = sorted(expand, key = lambda x : (-x[1],x[0]))
    for num in expand:
        answer += str(num[0])
    return str(int(answer))

맞춘 풀이

그러면 1214 vs 12 를 정상적으로 수행하기 위해서 1214 vs 1212면 어떨까 하고 아래와 같이 expandedNum을 만드는 과정을 수정했더니 정상적으로 해결이 되었다.

def solution(numbers):

    expand = []
    answer = ''
    for number in numbers:
        length = len(str(number))
        expandedNum = str(number)
        while len(expandedNum) <4:
            expandedNum += expandedNum
        expandedNum = expandedNum[:4]
        expand.append([number, int(expandedNum)])
    expand = sorted(expand, key = lambda x : (-x[1],x[0]))
    for num in expand:
        answer += str(num[0])
    return str(int(answer))