출처 : https://www.acmicpc.net/problem/2437
문제
하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 한다. 이 저울의 양 팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있다.
무게가 양의 정수인 N개의 저울추가 주어질 때, 이 추들을 사용하여 측정할 수 없는 양의 정수 무게 중 최솟값을 구하는 프로그램을 작성하시오.
예를 들어, 무게가 각각 3, 1, 6, 2, 7, 30, 1인 7개의 저울추가 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21이다.
입력
첫 째 줄에는 저울추의 개수를 나타내는 양의 정수 N이 주어진다. N은 1 이상 1,000 이하이다. 둘째 줄에는 저울추의 무게를 나타내는 N개의 양의 정수가 빈칸을 사이에 두고 주어진다. 각 추의 무게는 1이상 1,000,000 이하이다.
출력
첫째 줄에 주어진 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 출력한다.
아이디어
False로 이루어진 is_valid 리스트를 만든다.
만들 수 있는 무게는 permutation으로 무게 추를 뽑은 뒤, sum.
해당 무게의 인덱스를 True로 초기화
첫 번째 False의 인덱스 출력
코드
Mine (메모리 초과)
import sys
from itertools import permutations
input = lambda : sys.stdin.readline()
N = int(input())
weight = sorted(list(map(int, input().split())))
is_valid = [False] * (sum(weight) + 1)
is_valid[0] = True
min = 0
for i in range(len(weight)):
permut = list(permutations(weight, i))
for case in permut:
is_valid[sum(case)] = True
if min < is_valid.index(False):
min = is_valid.index(False)
else:
break
print(min)개선사항
is_valid (memorization) 방식 없앤다.
Clone
n = int(input())
s = list(map(int, input().split()))
s.sort()
num = 1
for i in range(n):
if num < s[i]:
break
num += s[i]
print(num)
