본문 바로가기
알고리즘?/기본2

3. k번째 큰수 set()

by 몰라닉네임 2023. 5. 4.

새롭게 알게 되는 함수 

 

  • set()

 중복을 제거하는 자료구조이다. 

 set() 자료구조는 sort 기능이 없어 다시 list 화 시켜줘야 한다.  ex) res=list(res)

 add (o )!! append(X)


[문제 k번째 큰수]

 

현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다. 현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려고 합니다. 기록한 값 중 k번째로 큰 수를 출력하는 프로그램을 작성하세요. 

만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19....이고 k값이 3이라면 k번째 큰 값은 22입니다.

 

-입력 설명 

첫 줄에 자연수 N(2<=N<=100)과 K(1<=K<=50)입려되고, 그 다음 줄에 N개의 카드값이 입력된다.

 

-출력설명 

첫 줄에 K번째 수를 출력합니다. 

 

-입력예제1

10 3 

13 15 34 23 45 65 33 11 26 42

 

-출력예제 1

143

 

-풀이: 

 65 45 42 34 33 26 23 15 13 11  제일 큰수 ,, k번째로 합한 값 중 k번째로 큰수 

3장 씩 뽑아서 더한 값을 모두 다 리스트로 저장하고 중복되는 값만 주의 

합해서 3번째로 큰수 65+45 =110 (42 34 33)


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

#띄어쓰기로 두개의 데이터를 입력받
n, k=map(int, input().split())

a=list(map(int, input().split()))


# 중복을 제거하는 자료구조 set(), res에 중복된 값을 제거해준다.
res=set()

#삼중 포문 (세개 데이터 접근, 선택)
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])
            
res=list(res) #set() 자료구조는 sort기능이 없다. 그래서 다시 리스트화 해준다.
res.sort(reverse=True) 
print(res[k-1])

 

*본 문제는 인프런 김태원 강사님의 파이썬 알고리즘 문제풀이 강의에 기반합니다.  

 

'알고리즘? > 기본2' 카테고리의 다른 글

8. 뒤집은 소수  (1) 2023.05.06
6. 자릿수의 합  (0) 2023.05.05
5. 정다면체  (0) 2023.05.05
4. 대표값 eumerate()  (1) 2023.05.04
2. k번째 수  (0) 2023.05.04