파이썬 알고리즘 문제풀이 : Section 2-3 / k번째 큰 수

---

문제

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

---

내 풀이

# 80 점, 타임리밋
#import sys
#sys.stdin = open("index.txt", "rt")
n, k = map(int, input().split())
cards = list(map(int, input().split()))
sum = 0
li = []
for x in range(n):
    for y in range(n):
            for z in range(n):
                     if x != y and y != z and x != z:
                         sum = cards[x]+cards[y]+cards[z]
                         if sum not in li:
                             li.append(sum)
li.sort(reverse = True)
print(li[k-1])

 

점수

---

강사님 풀이

# 강사님 코드
#import sys
#sys.stdin = open("index.txt", "rt")
n, k = map(int, input().split())
a = list(map(int, input().split()))
res = set() #중복불가능한 자료구조 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) # set 은 sort 기능 없으므로 리스트로 바꾸기
res.sort(reverse=True)
print(res[k-1])

 

# set 자료구조 : 중복 없음
# j 는 i+1 부터, m 은 j+1 부터 돌아야 중복되지 않고 모든 카드 뽑을 수 있다

# 모두 0 부터 돌면 중복된다