[알고리즘][2] 가장 큰 수
2023. 8. 10. 14:44ㆍ알고리즘 풀이
문제 설명
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일 테스트케이스가 추가되었습니다.
나의 풀이
- 정렬 문제 같아 보이지만 사실상 개념은 그리디
- 왜냐하면 최선의 선택을 구현할 수 있기 때문 그리고 최선의 선택은 아이디어를 통해서 알 수 있기 때문
def solution(numbers):
# 가장 큰 수이기 때문에 그리디, 탐색, DP
# 탐색의 경우 완전탐색을 사용해야하는데 그럴 경우 100,000!을 봐야하기 때문에 PASS
# DP의 경우에도 이전의 연산값을 활용할 방법이 없기 때문에 PASS
# 그리디가 가능성 있어 보임. for문 한번에 찾을 수도 있기 때문
# 일단 9가 대장 98은 그 다음이 되겠지
# 이렇게 만드려면 정렬 근데 string으로 바꿔서
# 이렇게 하면 안됨
# 자릿수가 다른 걸 해결해야한다
# 98과 987 98987 > 98798
str_numbers = []
for idx, number in enumerate(numbers):
str_number = str(number)
i = 0
while len(str_number) <= 3:
str_number += str_number[i]
i += 1
str_numbers.append((str_number, idx))
sorted_str_numbers = sorted(str_numbers, reverse=True)
answer = ""
for number, idx in sorted_str_numbers:
answer += str(numbers[idx])
return str(int(answer))
배운 풀이
- functools의 cmp_to_key 함수를 활용한 풀이
- 사실상 정렬 문제에서의 유연성을 확보하는 데 중요한 함수
- return이 -1 또는 0 이면 그대로
- return이 1이면 위치가 바뀜
import functools
def solution(numbers):
def comparator(a, b):
c1 = int(a + b)
c2 = int(b + a)
return (c1 < c2) - (c1 > c2)
str_numbers = map(str, numbers)
sorted_str_numbers = sorted(str_numbers, key=functools.cmp_to_key(comparator))
answer = ''.join(sorted_str_numbers)
return str(int(answer))
Reference
https://school.programmers.co.kr/learn/courses/30/lessons/42746/solution_groups?language=python3
'알고리즘 풀이' 카테고리의 다른 글
[알고리즘][2][X] 더 맵게 (0) | 2023.08.12 |
---|---|
[알고리즘][2] H-Index (0) | 2023.08.11 |
[알고리즘][2] 다리를 지나는 트럭 (0) | 2023.08.09 |
[알고리즘][2][X] 기능개발 (0) | 2023.08.08 |
[알고리즘][X] 방의 개수 (0) | 2023.08.07 |