[구현력 기르기] -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

 

 

나의 코드

n,k=map(int,input().split())
arr=list(map(int,input().split()))

res=set()

for i in range(n):
    for j in range(i+1,n):
        for q in range(j+1,n):
            s=arr[i]+arr[j]+arr[q]
            res.add(s)

res=list(res)
res.sort(reverse=True)

print(res[k-1])

중복이 포함된 n개의 숫자들중 3개를 골라서 더한 값들의 중 K 번째 값을 찾는 문제이다.

중복이 포함된 n 개이기 때문에 그저 3개를 더하면 중복된 값으로 인해 k 번째 인덱스를 바로 못찾기 때문에 집합 자료구조를 사용하였다.

 

또한 집합 자료에는 정렬함수인 sort를 사용하지 못하여 list로 변환한 한 뒤 정렬을 진행 하였다.

 

 

정답코드

import sys
sys.stdin=open("input.txt", "r")
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])

나와 거의 완전히 동일하다 !