包子凑数问题(完全背包问题+不定方程的知识)
题目描述
小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有N种蒸笼,其中第i种蒸笼恰好能放Ai个包子。每种蒸笼都有非常多笼,可以认为是无限笼。
每当有顾客想买X个包子,卖包子的大叔就会迅速选出若干笼包子来,使得这若干笼中恰好一共有X个包子。比如一共有3种蒸笼,分别能放3、4和5个包子。当顾客想买11个包子时,大叔就会选2笼3个的再加1笼5个的(也可能选出1笼3个的再加2笼4个的)。
当然有时包子大叔无论如何也凑不出顾客想买的数量。比如一共有3种蒸笼,分别能放4、5和6个包子。而顾客想买7个包子时,大叔就凑不出来了。
小明想知道一共有多少种数目是包子大叔凑不出来的。
第一个例子:
2
4
5
输出:
6
输入
第一行包含一个整数N。(1 <= N <= 100)
以下N行每行包含一个整数Ai。(1 <= Ai <= 100)
输出
一个整数代表答案。如果凑不出的数目有无限多个,输出INF。
样例输入
2
4
6
样例输出
INF
提示
对于样例1,凑不出的数目包括:1, 2, 3, 6, 7, 11。
对于样例2,所有奇数都凑不出来,所以有无限多个。****
思路
题目说明了每笼包子的个数,但是每种蒸笼的笼子是无限的,既由此想到完全背包问题。如果要凑出包子数,则要满足a1x1+a2x2+a3x3+…+anxn=c,这类不定方程。
有关不定方程的知识(相关证明省略)
假设不定方程为ax+by=c:
如果a,b互质,那么x,y一定有解且有无穷多个。但当x,y>=0时,ax+by=c此时导致方程无解c的个数有限,且max©=a*b-a-b。
如果a,b不互质,则不能保证有解,当且仅当c为a,b公约数(除1外)的倍数时才有解,则说明在a,b不互质的情况下有无穷多个c使得方程无解。
从上述二维的式子,我们可以推广到n维的式子去,如下:
a1x1+a2x2+a3x3+…+anxn=c,如果a1,a2,a3…an互质,则x1,x2,x3…xn一定有解且有无穷多个。如果a1,a2,a3…an不互质,那么就有无穷多个c使得方程无解。
递推关系式
首先先定义一个数组dp[i],表示当包子数i时,能否凑出,其值为0或1,当顾客想买的包子数为0时,那么包子大叔是肯定能凑出来的,所以可以先定义边界dp[0]=1,在包子数为0的基础上,包子数加2,即包子数为2时,也是能凑出来的,因此dp[0+2]=1,在包子数为2的基础上,包子数加3,即包子数为5时,也是能凑出来的,即dp[2+3]=1。就是说如果当包子数为i时,包子大叔能凑出来,那么在i的基础上加入输入另一笼的包子数a[i]也是可以凑出来的。由此,可以得到递推关系式:
if(dp[i]) dp[i+a[i]]=1;
代码展示
#include <iostream>
using namespace std;
int gcd(int a,int b)//运用递归方法求最大公约数
{
if(b==0) return a;
return gcd(b,a%b);
}
int main()
{
int n,b,i,j;
int a[101];
int ans;
cin>>n;
int dp[10000]={0};
dp[0]=1;
for(i=1;i<=n;i++)
{
cin>>a[i];//每笼的包子数
if(i==1) b=a[i];//初始化最大公约数
else
b=gcd(a[i],b);
}
if(b!=1) //判断是不是互质
{
cout<<"INF"<<endl;
}
else
{
for(i=1;i<=n;i++)
{
for(j=0;j<10000;j++)
{
if(dp[j])
{
dp[j+a[i]]=1;
}
}
}
ans=0;
for(i=0;i<10000;i++)
{
if(dp[i]==0)
ans++;
}
cout<<ans;
}
return 0;
}