2437. 저울
문제
하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 한다. 이 저울의 양 팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있다.
무게가 양의 정수인 N개의 저울추가 주어질 때, 이 추들을 사용하여 측정할 수 없는 양의 정수 무게 중 최솟값을 구하는 프로그램을 작성하시오.
예를 들어, 무게가 각각 3, 1, 6, 2, 7, 30, 1인 7개의 저울추가 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21이다.입력
첫 째 줄에는 저울추의 개수를 나타내는 양의 정수 N이 주어진다. N은 1 이상 1,000 이하이다. 둘째 줄에는 저울추의 무게를 나타내는 N개의 양의 정수가 빈칸을 사이에 두고 주어진다. 각 추의 무게는 1이상 1,000,000 이하이다.
출력
첫째 줄에 주어진 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 출력한다.
import sys
input = sys.stdin.readline
n = int(input())
weights = sorted(list(map(int, input().split())))
comp = 1
for weight in weights:
if comp >= weight:
comp += weight
else:
break
print(comp)
이 문제와 거의 비슷한 문제이다.
한번 복습해서 그런지 dp로 풀어볼까 하다가 갑자기 풀이가 생각이 났다.
comp < weight
조건일 때 반복문을 종료하면 된다.
comp 보다 적은 무게는 모두 비교할 수 있기 때문에 weight + comp까지 무게 측정가능하다.
예를 들어보자
comp = 5 이면 무게가 1, 2, 3, 4인 물건은 비교가 가능하다는 것이다.
weight = 3이라고 하자 👉 3 + 3, 4 + 3의 무게까지 비교가 가능해진다
만약 weight = 6라고 하면 무게가 5인 물건을 측정할 수가 없게 된다.
즉 무게가 comp인 물건은 측정할 수가 없다.
👉 comp < weight