문제:
숫자의 신은 여러명이 있지만, 그 중에 자연수의 신은 오세준이다. 오세준은 자연수의 신으로 오래오래 살다가 어느 날 음수의 신과 전쟁을 하게 되었다. 오세준은 음수의 신 이다솜을 이기기위해서 큰 숫자를 만들기로 했다.
오세준은 지금 K개의 자연수를 가지고 있다. 오세준은 K개의 수 중에 정확하게 N개의 수를 뽑아내서 그 수를 붙여서 만들 수 있는 수중에 가장 큰 수를 만들고자 한다. 같은 수를 여러 번 이용해도 된다. 단, 모든 수는 적어도 한 번은 이용되어야 한다.
오세준이 현재 가지고 있는 K개의 수가 주어졌을 때, 이 수를 적어도 한 번 이상 이용해서 만들 수 있는 수 중에 가장 큰 수를 출력하는 프로그램을 작성하시오.
예를 들어 지금 오세준이 2, 3, 7 이라는 3개의 수를 가지고 있고, 4개의 수를 뽑아야 한다면, 7732를 만들면 가장 큰 수가 된다.
입력:
첫째 줄에 K와 N이 공백을 사이에 두고 주어진다. K와 N은 각각 1,000보다 작거나 같은 자연수이고, N은 K보다 크거나 같다. 둘째 줄에는 K개의 수가 한 줄에 하나씩 주어진다. 각 수는 1,000,000,000보다 작거나 같은 자연수이다. 이 수는 중복되어 들어올 수 있다. 만약 중복되어서 들어오는 경우에는 각 수가 적어도 입력으로 주어진 만큼 이용되어야 한다는 소리다.
출력:
N개의 수를 뽑아서 연결해서 만들 수 있는 가장 큰 수를 출력한다.
풀이방법:
가지고 있는 수 K 보다 사용해야 하는 갯수 N이 더 크면 가장 큰 숫자를 여러번 사용하는 것이 큰 수를 만드는데 유리하다. 따라서 가장 큰 수를 찾은 뒤에 N-K개 만큼 배열에 추가적으로 넣어준다.
큰 수를 만들기 위해서는 두 수를 앞 뒤로 붙여보면 된다. 예를 들어 100 과 9 두 숫자가 있다고 했을 때, 큰 수가 무조건 가장 앞으로 오게 한다면 1009를 생성하게 되지만 실제로 큰 수는 9100가 되어야 한다. 따라서 1009 vs 9100과 같이 두 수를 번갈아 붙여보며 큰 수를 찾는 방법을 사용해야 한다.
따라서 python의 sort의 key 기능을 사용하며, 특수한 함수를 기준으로 정렬을 해야 하기 때문에 cmp_to_key를 사용하도록 한다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
from functools import cmp_to_key
def cmp(x,y):
if str(x)+str(y)>str(y)+str(x):
return -1
else:
return 1
k, n = map(int,input().split())
numbers = [int(input()) for _ in range(k)]
numbers = sorted(numbers,reverse=True)
for _ in range(k,n):
numbers.append(numbers[0])
numbers = sorted(numbers,key=cmp_to_key(cmp))
print(*numbers,sep='')
|
cs |
문제링크:
https://www.acmicpc.net/problem/1422
'Algorithm > Python' 카테고리의 다른 글
[BOJ]1024. 수열의 합 (0) | 2021.06.01 |
---|---|
[BOJ]1826. 연료 채우기 (0) | 2021.05.27 |
[BOJ]15686. 치킨 배달 (0) | 2021.05.20 |
[BOJ]2583. 영역 구하기 (0) | 2021.05.18 |
[BOJ]3055. 탈출 (0) | 2021.05.11 |