[프로그래머스] 코딩테스트 연습 - 가장 큰 수 (Python)

코딩테스트 연습 - 가장 큰 수 [Sort]

코딩테스트 연습 - 가장 큰 수

주어진 입출력 예의 두 번째 예제를 어떻게 풀까 고민하다가 해결했네요ㅎㅎ

 

풀이

먼저 제 풀이를 간단하게 설명하면 다음과 같습니다.

1. 모든 원소를 str으로 변환

2. (str원소*4)의 가장 앞 4자리를 기준으로 정렬

 

2번은 두 번째 예제 [3, 30, 34, 5, 9]를 예로 들어 설명해보겠습니다.

문제에서도 그렇고 해당 예제에서 중요한 점은 3, 30, 34처럼 "앞자리가 같은 숫자를 어떻게 정렬할 것인지"이겠죠.

먼저 3, 30, 34를 이용해 만들 수 있는 가장 큰 수는 34330입니다.결과를 잘 보면 가장 앞자리의 숫자 3을 기준으로 두 번째 자릿수가 3보다 큰 것은 앞으로 작은 것은 뒤로 정렬됩니다.결국 34, 3, 30 중에서 3은 3 자체로 정렬된 것이 아니라 33으로 정렬된 것과 마찬가지인 것이죠.

 

원소의 범위는 0~1000이니 1의 자리 숫자와 1000을 비교하기 위해 str으로 변환한 원소에 4배를 해서 앞자리 4개를 비교하면 문제를 해결할 수 있는 것입니다.

 

 

Code

def solution(numbers):
    numbers = sorted(list(map(str, numbers)), key=lambda k: (k*4)[:4], reverse=True)
    return str(int(''.join(numbers)))

 

 

문제 설명

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	resturn
[6, 10, 2]	"6210"
[3, 30, 34, 5, 9]	"9534330"

 

 

 

 

 

 

 

*