(인프런) 파이썬 알고리즘 문제1-3 "K번째 큰 수"

python

Problem

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

 

💯 Solution ①

: Only My Thinking

# 파일에서 값을 가져옴
import sys
# 채점을 할 때는 아래 코드 주석 처리
sys.stdin=open("input.txt", "rt")
 
# n개의 정수와 K번째 큰 값을 가져온다.
n, k =map(int, input().split())
# n개의 리스트를 리스트형식으로 가져온다.
n_list=list(map(int,input().split()))
 
result=[]
 
for i in range(n):
    for j in range(i+1,n):
        for m in range(j+1,n):
            result.append(n_list[i]+n_list[j]+n_list[m])
 
result = list(set(result))
result.sort(reverse=True)
 
print(result[k-1])

 

💯 Solution ②

: Tutor advice

# 파일에서 값을 가져옴
import sys
# 채점을 할 때는 아래 코드 주석 처리
sys.stdin=open("input.txt", "rt")
 
# n개의 정수와 K번째 큰 값을 가져온다.
n, k =map(int, input().split())
# n개의 리스트를 리스트형식으로 가져온다.
a=list(map(int,input().split()))
 
res=[]
 
# 3중 for문 : 주어진 배열에서 3개를 뽑는 경우의 수 구현
for i in range(n):
    for j in range(i+1,n):
        for m in range(j+1,n):
            # 중복을 제거하면 res에 입력
            res.append(a[i]+a[j]+a[m])
 
# 중복을 제거하는 자료구조 : set()
# res를 list화 시켜서 sort사용(내림차순 정렬)
res=list(set(res))
res.sort(reverse=True)
# k번째 추출
print(res[k-1])