题目链接
题目描述
小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有 N N N 种蒸笼,其中第 i i i 种蒸笼恰好能放 A i A_i Ai 个包子。每种蒸笼都有非常多笼,可以认为是无限笼。
每当有顾客想买 X X X 个包子,卖包子的大叔就会迅速选出若干笼包子来,使得这若干笼中恰好一共有 X X X 个包子。比如一共有 3 种蒸笼,分别能放 3、4 和 5 个包子。当顾客想买 11 个包子时,大叔就会选 2 笼 3 个的再加 1 笼 5 个的(也可能选出 1 笼 3 个的再加 2 笼 4 个的)。
当然有时包子大叔无论如何也凑不出顾客想买的数量。比如一共有 3 种蒸笼,分别能放 4、5 和 6 个包子。而顾客想买 7 个包子时,大叔就凑不出来了。
小明想知道一共有多少种数目是包子大叔凑不出来的。
输入描述
第一行包含一个整数
N
(
1
≤
N
≤
100
)
N (1 \leq N \leq 100)
N(1≤N≤100)。
以下
N
N
N 行每行包含一个整数
A
i
A_i
Ai
(
1
≤
A
i
≤
100
)
(1 \leq A_i \leq 100)
(1≤Ai≤100)
输出描述
一个整数代表答案。如果凑不出的数目有无限多个,输出 "INF"
。
输入输出样例
输入
# -*- coding: utf-8 -*-
# @Author : BYW-yuwei
# @Software: python3.8.6
2
4
5
输出
6
样例说明:
凑不出的数目包括:1, 2, 3, 6, 7, 11。
解题思路
数学+动态规划
裴署定理:x,y的最大公约数为d,那么ax+by(a,b均为整数)任意组合均为d的倍数。
任务需求:对
n
n
n 个蒸笼的容量进行线性组合,看各种组合能否覆盖自然数域。
如果本题输入的数据的最大公约数不等于1,那么凑不出的数目必定有无限多个。
即通过
g
c
d
gcd
gcd 来判断输入的
n
n
n 个蒸笼大小对自然数域的覆盖情况。
1、定义一个列表 capy 记录
n
n
n 个蒸笼大小;
2、循环遍历,计算出这
n
n
n 个数字共同的
g
c
d
gcd
gcd ,记为
g
g
g 。
3、如果
g
=
1
g=1
g=1 ,那么这个组合可以在整数的定义域内完成对所有整数的覆盖,否则,该组合将不能完成对整数的覆盖。因此如果
g
≠
1
g\ne 1
g=1 ,那么可以直接输出 "INF"
。
4、即使
g
=
1
g=1
g=1 ,对部分较小的整数,当前组合也未必能覆盖。这时,该问题变成一个动态规划问题。定义一个列表 dp 存储 0,1, dp
[
i
]
[i]
[i] 表示当前组合是否能够覆盖自然数
i
i
i 。
(互质数a,b的不能凑出来的最大数为 a*b-a-b )
数组长度最多100个,最大的数不超过100,由因为他们是互质数,所以不能凑出来的最大数为100*99-100-99,因此 dp 列表的长度设置到
10
0
2
100^2
1002 即可,超过该范围的正整数一定能够被覆盖。
0 一定能够被覆盖,因此 dp [ 0 ] [0] [0]=1 。接着对每个蒸笼进行递推,对于每一个递推,遍历 1 0 4 10^4 104 长度的区间,将能够有组合构成的数 j j j 的 dp [ j ] [j] [j] 设置为 1 。完成递推后,再次遍历 dp 列表,统计列表中 0 的数量即为结果。
AC Code
Python3
import sys
n=int(input())
capy=[0]*n
for i in range(n):
capy[i]=int(input())
def gcd(x:int,y:int):
if x<y:x,y=y,x
while y:
t=x%y
x=y
y=t
return x
lim_flag=False
g=capy[0]
for i in range(n):
g=min(g,gcd(g,capy[i]))
if g==1:
lim_flag=True
break
if not lim_flag:
print("INF")
sys.exit(0)
N=int(1e4)
dp=[0]*(N+1)
dp[0]=1
for i in range(n):
for j in range(0,N+1):
if j+capy[i]>N:break
if dp[j]:
dp[j+capy[i]]=1
print(dp.count(0))