파이썬 알고리즘-3 k번째 큰 수

jiffydev·2020년 8월 18일
0

Algorithm

목록 보기
3/134
post-thumbnail

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'이 되려면 같은 합 결과의 경우는 다른 번째라고 간주하지 않습니다.

내 코드

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])

보통 알고리즘은 수행시간도 고려해야되는데 이 경우 for문이 3중첩이라 이렇게 해도 되는지 계속 고민했다.(100개밖에 안되니 상관없을지도?)
또 각 반복문에서 마지막 index를 -2, -1 해줘야 하지 않나 싶은데 풀이에서는 그냥 N으로 놓고 푼 이유를 모르겠다ㅠ▶ 확인한 결과 둘 다 상관 없음. range(start, end, step)는 start>step이면 알아서 빈 배열로 돌려주기 때문

풀이

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])

반성점

  • 수행시간이 느려보여도 일단 코드 치자

배운 것

  • N개를 묶어서 순서대로 더할 때는 반복문의 시작을 직전 반복문의 시작 인덱스+1로 해줘야 한다. j+1을 i+2로 했다가 답 안나옴
  • list.sort(reverse=True)로 내림차순 정렬이 가능

생각해볼 것

  • 저번에 봤던 코딩테스트에서는 set을 쓰지 말라고 명시해둔 곳이 있었는데, 이럴 경우는 어떻게 처리할지
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글