[알고리즘] 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번째 수는 반드시 존재합니다.
테스트케이스
입력예제 | 출력예제 |
---|---|
10 3 13 15 34 23 45 65 33 11 26 42 |
143 |
10 3 18 54 46 52 28 22 23 53 28 40 |
152 |
해결방법
3중 반복문을 이용한다.
코드
n, k = map(int, input().split())
cards = list(map(int, input().split()))
cards.sort(reverse=True)
sums = set()
for i in range(n-2):
for j in range(i+1, n-1):
for z in range(j+1, n):
sums.add(sum([cards[i], cards[j], cards[z]]))
print(sorted(sums, reverse=True)[k-1])
댓글남기기