티스토리 뷰

Test/Python(20220101~)

K번째 큰 수

kiostory 2023. 7. 23. 13:38
현수는 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

 

 

import sys
sys.stdin=open("input.txt", "rt", encoding='UTF8')

N, K = map(int, input().split())
Card = list(map(int, input().split()))
L=[]

for i in range(len(Card)-2):
    for j in range(i+1,len(Card)-1):
        for h in range(j+1,len(Card)):
            L.append(Card[i]+Card[j]+Card[h])

L.sort(reverse=True)
L=list(dict.fromkeys(L)) #<----- 중복제거
print(L[K-1])

Sort 전 경우의 수 : 
 [62, 51, 73, 93, 61, 39, 54, 70, 70, 92, 112, 80, 58, 73, 89, 81, 101, 69, 47, 62, 78, 123, 91, 69, 84, 100, 111, 89, 104, 120, 57, 72, 88, 50, 66, 81, 72, 94, 114, 82, 60, 75, 91, 83, 103, 71, 49, 64, 80, 125, 93, 71, 86, 102, 113, 91, 106, 122, 59, 74, 90, 52, 68, 83, 102, 122, 90, 68, 83, 99, 144, 112, 90, 105, 121, 132, 110, 125, 141, 78, 93, 109, 71, 87, 102, 133, 101, 79, 94, 110, 121, 99, 114, 130, 67, 82, 98, 60, 76, 91, 143, 121, 136, 152, 89, 104, 120, 82, 98, 113, 109, 124, 140, 102, 118, 133, 70, 86, 101, 79]


Sort 및 중복제거 후 : 
 [152, 144, 143, 141, 140, 136, 133, 132, 130, 125, 124, 123, 122, 121, 120, 118, 114, 113, 112, 111, 110, 109, 106, 105, 104, 103, 102, 101, 100, 99, 98, 94, 93, 92, 91, 90, 89, 88, 87, 86, 84, 83, 82, 81, 80, 79, 78, 76, 75, 74, 73, 72, 71, 70, 69, 68, 67, 66, 64, 62, 61, 60, 59, 58, 57, 54, 52, 51, 50, 49, 47, 39]

 

그래서 예제의 정답은,
143

 

 

 

cf. set()

import sys
sys.stdin=open("input.txt", "rt", encoding='UTF8')

N, K = map(int, input().split())
Card = list(map(int, input().split()))
L=set()   #<--------중복제거하는 변수 L

for i in range(len(Card)-2):
    for j in range(i+1,len(Card)-1):
        for h in range(j+1,len(Card)):
            L.add(Card[i]+Card[j]+Card[h])  #<--------set()은 append가 아니고 add!

L=list(L) #<-------- set()은 sort()가 불가하므로 다시 list로.   
L.sort(reverse=True)
print(L[K-1])
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/08   »
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 26 27 28 29 30
31
글 보관함