[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 합니다.

입출력 예

 

💡 풀이

# 처음 풀이
def solution(numbers):
    answer = ''
    
    numbers = list(map(str, numbers)) # numbers 배열의 원소 string으로 변환
    numbers.sort(key = lambda x:x*3, reverse=1) # 정렬
    for i in numbers:
        answer += i  # answer에 하나의 string 저장
    
    return str(int(answer)) # 0인 경우 고려해 return

# 짧은 코드
def solution(numbers):
    ans = sorted(list(map(str, numbers)), key = lambda x:x*3, reverse=1)
    
    return str(int(''.join(ans)))

 

알고리즘

• 배열에 존재하는 원소는 0이상 1000미만 (최대 3자리수)
  • string의 3배 할 경우 (EX. 3→333, 340→340340) 정렬시 사전순서로 정렬됨
  • lambda 함수 이용해 string 3배했을 때를 기준으로 reverse 정렬(큰 순서대로)
• 배열 내의 원소 합쳐 하나의 string으로 변환
• 반환된 string이 0인 경우 고려해 int로 변환한 후 string으로 다시 변환해 return