Programming Language/Python

Python Algorithm) 3. k번째 큰 수

HJChung 2020. 4. 3. 14:31

3. 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번째 수는 반드시 존재합니다.

 

입력예제 1
10 3
13 15 34 23 45 65 33 11 26 42

 

출력예제 1

143

 

'같은 숫자의 카드가 여러장 있을 수 있습니다.' 같이 중복이슈가 있는 경우, 문제 해석에 집중해야 하는 문제입니다. 실수가 많을 수 있습니다. 

또한 '만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값 은 22입니다.'이 예시의 해석도 중요합니다. 'K값이 3이라면 K번째 큰 값 은 22'이 되려면 같은 합 결과의 경우는 다른 번째라고 간주하지 않습니다. 

 

python 문제 풀이시 '중복이슈'가 등장하는 경우
중복을 제거하는 set() 자료구조가 있다는 것을 기억합니다. 
# set(): 중복되는 값은 딱 한 번만 저장하는, 즉 중복을 제거하는 자료구조 입니다. 단, set자료구조는 sort() 기능이 없으므로, 다시 list로 만들어 준 뒤, sort()를 취해주어야 합니다. 

# python 문제 풀이시 '중복이슈'가 등장하는 경우
# 중복을 제거하는 set() 자료구조가 있다는 것을 기억합니다. 
# set(): 중복되는 값은 딱 한 번만 저장하는, 즉 중복을 제거하는 자료구조 입니다. 단, set자료구조는 sort() 기능이 없으므로, 다시 list로 만들어 준 뒤, sort()를 취해주어야 합니다. 
# 이때, 내림차순으로 정렬해야 '[k-1]이 k번째로 큰 수'가 되므로 
# sort(reverse = True)로 해 줍니다. 

N, k = map(int, input().split())
arr = list(map(int, input().split())

result = set() #중복되는 것은 한 번만 저장하는 자료구조
result = [arr[i]+arr[j]+arr[s] for i in range(N) for j in range(i+1, N) for s in range(j+1, N)]

result = list(result)
result.sort(reserve = True)

print(result[k-1])