PCCP 시험 준비하면서 인상깊었던 문제를 정리한 글입니다.
문제 설명
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" |
코드
def solution(numbers):
numbers = list(map(str, numbers))
numbers.sort(key=lambda x: x * 3, reverse=True)
result = ''.join(numbers)
return str(int(result))코드 설명
주어지는 정수들을 이어붙여서 가장 큰 수를 만드는 문제이다.
모두 같은 자릿수의 수가 주어지는 것이 아니므로 잘 생각해보아야 풀 수 있다. "입출력 예"에서 2번째 테스트케이스를 보자.
numbers.sort(reverse=True)[3, 30, 34, 5, 9]를 단순히 문자열로 형변환하고 역순으로 정렳하면 9, 5, 34, 30, 3으로 정렬되어 9534303이 반환되는데, 이는 정답이 아니다.
이 경우가 바로 자릿수를 고려하지 않은 것이다.
numbers.sort(key=lambda x: x * 3, reverse=True)위의 로직으로 문자열을 각각 3씩 곱한 문자열을 기준으로 역순으로 정렬해보자. 그러면 9, 5, 34, 3, 30이 되어 9534330이 반환된다. 이렇게 가장 큰 수를 도출할 수 있다.
자릿수가 다른 수들을 가지고 가장 큰 수를 만드려면 다음을 고려해야 한다.
- 주어지는 원소는 0 이상 1,000 이하이다.
- 숫자가 최대 3자리이기 때문에 원소를 3번씩 붙인 문자열을 정렬하면 된다. (1,000은 4자리이긴 하지만 상관없다.)
🌱