코딩테스트/Python
2. 자료구조와 알고리즘 입문 03.K번째 큰 수
star빛
2022. 7. 11. 17:10
03.K번째 큰 수
import sys
sys.stdin=open("/Users/eomminjeong/codingtest/pythonalgorithm/섹션 2/3. k번째 큰 수/in1.txt","rt")
n,k=map(int, input().split())
a=list(map(int, input().split()))
res=set()
#set은 중복을 제거하는 자료구조
# 3개의 자료를 뽑아서 합한 후 res 에 추가
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은 append 아니고 add
#set은 sort 없어서 list 화 시켜야함.
res=list(res)
res.sort(reverse=True) #reverse=True 내림차순
print(res[k-1])
- 3개를 뽑을때 2중 3중 포문은 i+1 j+1 부터 시작해야함.
- set은 append 아니고 add 해야함.
- set은 중복을 제거하는 자료구조.
- set은 sort없어서 list 로 바꾸어주어야 함.