백준 온라인 저널, 그리디 알고리즘/2437번 : 저울(파이썬)

문제 정의

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

 

 

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

 

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

 

 

입력

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

 

 

출력

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

 

 

예제 입력 1

7
3 1 6 2 7 30 1

예제 출력 1

21

 

 

접근 방법

1. 주어진 추의 무게를 오름차순으로 정렬한다.

2. total이라는 변수에 추의 무게 중 첫 원소(가장 가벼운)를 입력하고, 두 번째에 있는 원소부터 차례로 뽑는다. 이때 total보다 현재 뽑힌 원소 - 1보다 작을 경우 뽑는 과정을 멈춘다.

2-1. 만약 추의 무게 중 첫 원소가 1보다 크다면 1을 출력한다.

3. 이후 total + 1을 출력한다.

 

코드

n = int(input())
array = list(map(int, input().split()))
array.sort()
if array[0] == 1:
    total = array[0]
    for i in range(1, len(array)):
        if total < array[i]-1:
            break
        else:
            total += array[i]
    print(total+1)
else:
    print(1)