본문 바로가기

백준 그리디 알고리즘

백준 그리디 알고리즘 2437 저울

문제

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

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

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

입력

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

출력

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

풀이

주어진 추로 측정할 수 없는 가장 작은 수를 구해야 한다.

무게는 1이상 1,000,000이하 이므로 가장 작은 수의 기준을 1로 둔다. state=1

입력 받은 추 무게 정보 리스트를 오름차순으로 정렬한다.

반복문을 통해 가장 작은 수 부터 기준과 비교한다.

만약 가장 작은 추가 1보다 크다면 1이 측정할 수 없는 가장 작은 수이기에 바로  출력한다.

추가 1보다 크지 않으면 1은 측정할 수 있는것이다. 이후 state에 그 리스트 값을 더하여

그 값을 측정할 수 있는지 비교한다.

비교 중 만약 state보다 해당 인덱스의 추 무게가 더 크다면 그 state 무게는 측정할 수 없는것이다.

import sys
n = int(sys.stdin.readline().rstrip("\n"))
nums = list(map(int, sys.stdin.readline().rstrip("\n").split()))
nums = sorted(nums)
state=1
for num in nums:
    if(state<num):
        break
    else:
        state+=num
print(state)