[BaekJoon] 16496 큰 수 만들기 (Platinum 5) - Python

문제

 

음이 아닌 정수가 N개 들어있는 리스트가 주어졌을 때, 리스트에 포함된 수를 나열하여 만들 수 있는 가장 큰 수를 구하는 프로그램을 작성하시오.

---

입출력 예시

 

입력: 첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 리스트에 포함된 수가 주어진다. 수는 공백으로 구분되어져 있고, 1,000,000,000보다 작거나 같은 음이 아닌 정수 이다. 0을 제외한 나머지 수는 0으로 시작하지 않으며, 0이 주어지는 경우 0 하나가 주어진다.

출력: 리스트에 포함된 수를 나열하여 만들 수 있는 가장 큰 수를 출력한다. 수는 0으로 시작하면 안되며, 0이 정답인 경우 0 하나를 출력해야 한다.

 

입력예시1)

5
3 30 34 5 9

출력예시1)

9534330

 

입력예시2)

5
0 0 0 0 1

출력예시2)

10000

---

코드

 

N = int(input()) # 정수 개수 입력
num = [] # 숫자 저장할 리스트 생성
for i in sorted(list(map(int, input().split()))): # 정렬된 리스트에서 하나씩 가져옴
    s = str(i) # 문자열로 바꿈
    l = len(s) # 문자열 길이 구함
    while len(s) < 10: s += s[len(s) - l] # 문자열의 길이가 10이 될 때 까지 반복
    # 첫 문자열이 123이었다면? => 1231231231처럼 10자리가 될 때 까지 각 자리를 반복하여 생성
    # ex) 101 => 1011011011
    num.append((list(s), str(i))) # 숫자를 저장할 리스트에 추가
num.sort(key = lambda x : x[0], reverse = True) # num리스트를 정렬하는데, 정렬 기준인 key를 각 리스트의 0번째 인덱스로하여 정렬
# ex) num = [[0, 1], [3, 2], [2, 8]] => [[3, 2], [2, 8], [0, 1]]
result = "" # 결과 저장할 변수 생성
for i in num: result += i[-1] # 정렬한 숫자 리스트에서 원래의 값만 순서대로 추가
print(result if int(result) != 0 else 0) # 결과 출력

---

실행 화면

 

---

채점 결과