[Programmers] 가장 큰 수 python (정렬)

문제

문제 설명

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

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

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

제한 사항

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

입출력 예

 

접근 방법

• 빈 문자열로 수를 초기화 한다.
• 수의 목록을 answer(결과값)을 크게 만드는 것 우선으로 정렬한다.
• 목록에서 하나씩 꺼내어 현재 수에 이어 붙인다.
• 모든 수를 다 사용할 때 까지 반복한다.

수의 목록을 answer(결과값)을 크게 만드는 것 우선으로 정렬해야 하므로 다른 기준으로 정렬을 시도하는 것이 이 문제의 핵심이다 !!

 

"크게 만드는 수"의 기준
숫자를 이어붙여서 최대 크기가 1000이므로 4글자까지 끊어서 비교한다.

 

풀이

def solution(numbers):
    numbers = list(map(str, numbers))
    numbers.sort(reverse=True, key=lambda x: (x*4)[:4]) # 아스키 값으로 변환되어 정렬됨
    if numbers[0] == '0':
    	answer = '0'
    else:
    	answer = ''.join(numbers)

    return answer

0이 여러개 있는 경우 000 이런식으로 답이 나올 수 있기 때문에

예외에 대한 처리를 해주어야 한다.

 

복잡도 분석

• 리스트에 들어있는 원소들을 str로 만듦  : O(n)
• 정렬 : O(nlogn)
• 전체 알고리즘 : O(nlogn)

 

def solution(numbers):
    numbers = list(map(str, numbers)) # O(n)
    numbers.sort(reverse=True, key=lambda x: (x*4)[:4]) # O(nlong)
    if numbers[0] == '0': # 상수시간 
    	answer = '0'
    else:
    	answer = ''.join(numbers) # O(n)
 
    return answer