[파이썬 알고리즘] 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

🧞 풀이

풀이

'''
[자료구조와 알고리즘 입문]

K번째 큰 수

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

25, 25, 23, 23, 22, 20, 19... K값이 3이라면 K번째 큰 값은 22
'''

# 입력: 첫째 줄 n, k
# 입력: 둘째 줄 n개의 카드값 입력

# 출력: k번째 수 출력

from itertools import combinations                              # itertools 패키지에서 combinations 모듈 불러오기

n, k = map(int, input().split())                                # n, k 정수 할당
n_list = input().split()                                        # n_list에 숫자 입력하여 리스트로 할당

data = combinations(n_list, 3)                                  # combinations() 함수 통해, n_list의 요소 3개씩 묶은 조합 data에 할당
a_list = []                                                     # 정답을 담을 a_list 빈 리스트 할당

for i in data:                                                  # data의 각 요소 i 반복
    a_list.append(sum((int(i[0]), int(i[1]), int(i[2]))))       # 각 요소의 합 더하여 a_list에 할당

tmp = set(a_list)                                               # a_list를 set 자료구조로 변환하여 tmp에 할당
a_list2 = list(tmp)                                             # a_list2는 기존의 list를 set으로 변환했다가 다시 list로 변환한 것
a_list2.sort(reverse=True)                                      # a_list2 내림차순 정렬

print(a_list2[k-1])                                             # a_list2의 k-1 인덱스의 값 출력

 

다른 사람 풀이

# K번째 큰 수

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

# 중복을 제거하는 자료구조
res = set()

# 중복 방지
# 삼중 반복문
# set은 append()가 아닌 add()

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])

# set은 sort()가 없음
# list로 바꿔서 정렬해줘야 함
# sort(reverse=True) 내림차순 정렬
res = list(res)
res.sort(reverse=True)
print(res[k-1])