[그리디 알고리즘6] 백준 2437 저울 파이썬
본문 바로가기
Python/Python 코딩테스트

[그리디 알고리즘6] 백준 2437 저울 파이썬

by 쏠수있어ㅤ 2021. 7. 16.
반응형

 

백준 2437 저울

 

📜 문제 

하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 한다. 이 저울의 양 팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있다.

무게가 양의 정수인 N개의 저울추가 주어질 때, 이 추들을 사용하여 측정할 수 없는 양의 정수 무게 중 최솟값을 구하는 프로그램을 작성하시오.

예를 들어, 무게가 각각 3, 1, 6, 2, 7, 30, 1인 7개의 저울추가 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21이다.

 

🚩 입력 

첫 째 줄에는 저울추의 개수를 나타내는 양의 정수 N이 주어진다. N은 1 이상 1,000 이하이다. 둘째 줄에는 저울추의 무게를 나타내는 N개의 양의 정수가 빈칸을 사이에 두고 주어진다. 각 추의 무게는 1이상 1,000,000 이하이다.

 

🌞 출력

첫째 줄에 주어진 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 출력한다.

 

🥨문제 풀이 

이삼중 for문에 combinations 까지 쓰며 만들다가 시간초과가 나고 도저히 이건 이 길이 아니다 싶어서 답을 찾아보았다. 역시나 시간 제한 1초일만한 원리가 있었다..

추가 담겨있는 리스트 = arr를 sort()로 오름차순 정렬을 해주고 하나씩 더하며 해당 수를 arr 다음 요소 (다음 추)와 비교를 한다. 비교해서 다음 추가 여태 더했던 추들의 합보다 더 크다면 해당 값은 주어진 N개의 추들로 만들 수 없다. 

 

예를 들어 1 2 3 4 20 이라는 5개의 추가 주어졌을 때 

target = 1 을 넣고 

target =1 이 arr[0] (1)보다 같거나 크면 -> target + arr[0] => 2

target = 2가 arr[1] (2) 보다 같거나 크면 -> target + arr[1] => 4

target = 4  > arr[2] (3)  -> += 

target = 7 > arr[3] (4) -> +=

target = 11 < 20 ----> 11은 나올 수 없는 수 ! 

여태까지의 추의 합이 다음 배열의 요소보다 작으면 해당 합 + 1이 나올 수 없는 수이다. 

 

만약 1,2,3 배열이라면 모든 배열의 수를 합한 값의 +1 이 정답이 되고

5,5,10 과 같은 배열의 경우 측정할 수 없는 무게는 1이된다. 

 

 

🥕 정답 코드 

n = int(input())
arr = list(map(int,input().split()))
arr.sort()

target =1
for i in arr:
    if target<i:
        break
    target += i
print(target)

 

 

반응형

댓글