[Python] K번째 큰 수

[문제]

현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다. 현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려 고 합니다. 3장을 뽑을 수 있는 모든 경우를 기록합니다. 기록한 값 중 K번째로 큰 수를 출력 하는 프로그램을 작성하세요. 만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값 은 22입니다.

 

▣ 입력설명

첫 줄에 자연수 N(3<=N<=100)과 K(1<=K<=50) 입력되고, 그 다음 줄에 N개의 카드값이 입력 된다.

 

▣ 출력설명

첫 줄에 K번째 수를 출력합니다. K번째 수는 반드시 존재합니다.

 

▣ 입력예제

1 10 3 13 15 34 23 45 65 33 11 26 42

 

▣ 출력예제

1 143

---

[풀이]

import sys

n, k = map(int, input().split())
a = list(map(int, input().split()))
res = set()

for i in range(n):
    for j in range(i + 1, n):
        for m in range(j + 1, n):
            res.add(a[i]+a[j]+a[m])

res = list(res)
res.sort(reverse=True)
print(res[k-1])

---

[중요내용]

1. 3중 for문은 총 n개의 숫자 중에서, 중복을 허용하지 않고 3개의 숫자를 모두 뽑는 코드이다.
2. 집합(set) 자료형 특징
   1. 중복 비허용
   2. 데이터를 추가 시: add() 함수 사용 (append() 함수 사용x)
   3. 순서가 없기 때문에, sort() 함수를 사용해서 정렬 불가
   4. 집합 자료형의 데이터를 정렬시키기 위해서는 반드시 집합 자료형을 리스트 자료형으로 형 변환한 후에, sort() 함수를 사용해서 정렬을 해야한다.
3. sort(reverse=True) 코드는 특정 자료형의 데이터를 내림차순(큰 수부터 작은 수로)으로 정렬하는 것을 의미한다.