动态规划：包子凑数

题目链接

蓝桥云课 包子凑数

题目描述

小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有 $N$ 种蒸笼，其中第 $i$ 种蒸笼恰好能放 $A_i$ 个包子。每种蒸笼都有非常多笼，可以认为是无限笼。

每当有顾客想买 $X$ 个包子，卖包子的大叔就会迅速选出若干笼包子来，使得这若干笼中恰好一共有 $X$ 个包子。比如一共有 3 种蒸笼，分别能放 3、4 和 5 个包子。当顾客想买 11 个包子时，大叔就会选 2 笼 3 个的再加 1 笼 5 个的（也可能选出 1 笼 3 个的再加 2 笼 4 个的）。

当然有时包子大叔无论如何也凑不出顾客想买的数量。比如一共有 3 种蒸笼，分别能放 4、5 和 6 个包子。而顾客想买 7 个包子时，大叔就凑不出来了。

小明想知道一共有多少种数目是包子大叔凑不出来的。

输入描述

第一行包含一个整数 $N(1\le N\le 100)$。
以下 $N$ 行每行包含一个整数 $A_i$$(1\le A_i\le 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$ 个蒸笼的容量进行线性组合，看各种组合能否覆盖自然数域。
如果本题输入的数据的最大公约数不等于1，那么凑不出的数目必定有无限多个。
即通过 $gcd$ 来判断输入的 $n$ 个蒸笼大小对自然数域的覆盖情况。

1、定义一个列表 capy 记录 $n$ 个蒸笼大小；
2、循环遍历，计算出这 $n$ 个数字共同的 $gcd$ ,记为 $g$ 。
3、如果 $g=1$ ，那么这个组合可以在整数的定义域内完成对所有整数的覆盖，否则，该组合将不能完成对整数的覆盖。因此如果 $g\ne 1$ ,那么可以直接输出 "INF" 。
4、即使 $g=1$ ,对部分较小的整数，当前组合也未必能覆盖。这时，该问题变成一个动态规划问题。定义一个列表 dp 存储 0，1， dp$[i]$ 表示当前组合是否能够覆盖自然数 $i$ 。
（互质数a,b的不能凑出来的最大数为 a*b-a-b ）
数组长度最多100个，最大的数不超过100，由因为他们是互质数，所以不能凑出来的最大数为100*99-100-99，因此 dp 列表的长度设置到 $100^2$ 即可，超过该范围的正整数一定能够被覆盖。

0 一定能够被覆盖，因此 dp$[0]$=1 。接着对每个蒸笼进行递推，对于每一个递推，遍历 $10^4$ 长度的区间，将能够有组合构成的数 $j$ 的 dp$[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))