3. 한국정보올림피아드 알고리즘 연습_3. k번째 큰 수
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(j+1, N) for s in range(j+1, N)]
result = list(result)
result.sor(reserve = True)
print(result[k-1])
3. k 번째 큰 수
현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다.
현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려 고 합니다. 3장을 뽑을 수 있는 모든 경우를 기록합니다.
기록한 값 중 K번째로 큰 수를 출력 하는 프로그램을 작성하세요.
만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰값 은 22입니다.
*K값이 3이라면 K번째 큰값 은 22 : 이 말은 중복을 포함하지 않는다는 의미
set() : 중복제거
*만약 큰 수부터 만들어진 수 : 내림차순 정렬이 필요한 경우
*.sort(reverse = True)를 사용
▣ 입력설명
첫 줄에 자연수 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'이 되려면 같은 합 결과의 경우는 다른 번째라고 간주하지 않습니다.
n, k=map(int, input().split())
a=list(map(int, input().split()))
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)
res.sort(reverse=True)
print(res[k-1])
의문점
for문이 3중첩이라...?
배운 것
순서대로 더할 때 직전 반복문+1로 진행되어야 한다.
( j+1을 i+2)로 할 경우 답이 나올 수 없음.
n n
i+1 i+1
i+2(x) j+1
#3
N,K = map(int, input().split())
lst=list(map(int, input().split()))
sum_lst=set()
#set()중복 제거
#set.add()로 추가
for i in range(N-2):
for j in range(i+1, N-1):
for k in range(j+1, N):
sum_lst.add(lst[i]+lst[j]+lst[k])
sum_lst=list(sum_lst)
sum_lst.sort(reverse=True)
print(sum_lst[K-1])