CodingTest/인프런 (Algorithm)
(인프런) 파이썬 알고리즘 문제1-3 "K번째 큰 수"
Juni_K
2021. 2. 13. 11:47
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
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# 파일에서 값을 가져옴
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])
|
cs |
💯 Solution ②
: Tutor advice
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | # 파일에서 값을 가져옴 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]) | cs |