[蓝桥杯][历年真题] PREV-36 包子凑数 完全背包问题

PREV-36 历届试题 包子凑数

资源限制
时间限制：1.0s 内存限制：256.0MB

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

每当有顾客想买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行每行包含一个整数Ai。(1 <= Ai <= 100)
　　
输出格式
　　一个整数代表答案。如果凑不出的数目有无限多个，输出INF。
　　
样例输入
2
4
5

样例输出
6

样例输入
2
4
6

样例输出
INF

样例说明
　　对于样例1，凑不出的数目包括：1, 2, 3, 6, 7, 11。
　　对于样例2，所有奇数都凑不出来，所以有无限多个。
　　
数据规模和约定
　　峰值内存消耗（含虚拟机） < 256M
　　CPU消耗 < 1000ms

解题思路：

本题是完全背包问题，唯一难度就是何时为INF，何时为有限个。
根据扩展的欧几里得定理：一组数字互为质数，即公因数为1，那么不能凑到的数就有限；
如果一堆数的最大公约数不为1，那么他们不可以组成的数有无限个。
于是先求这些数的最大公约数，如果为1，则再进一步用完全背包方法求出有限个数。

因为笼数上限最多100种笼子，每个笼子最多装100个包子，所以上限为最多10000个包子

---

#include<iostream>
#include<algorithm> 
using namespace std;

#define MAX_NUM 20

int f[MAX_NUM];//保存拿i个物品的可能性 
//f[i]表示拿i个包子是否可能，如果可以则尝试多拿一笼，1表示可以，0表示不可以 
int temp[101];//保存每个笼子的包子容量 
int N;

int gcd(int a, int b)//求两个数的最大公约数 
{
	if (a % b == 0)
		return b;
	else
		return gcd(b, a % b);
}
int gcdAll()//多个数的最大公约数 
{
	int num = temp[0];
	for (int i = 1; i < N; i++)
	{
		num = gcd(num, temp[i]);
	}
	return num;
}

int main() {

	cin >> N;
	for (int i = 0; i < N; i++)
	{
		cin >> temp[i];
	}

	if (gcdAll() != 1)
	{
		cout << "INF";
		return 0;
	}

	fill(f, f + MAX_NUM, 0);

	f[0] = 1;
	for (int k = 0; k < N; k++)
	{
		for (int j = temp[k]; j <= MAX_NUM; j++) //j要>=temp[k] 
		{
//			printf("f[%d-%d]=%d ",j,temp[k],f[j - temp[k]] );
			if (f[j - temp[k]] == 1){
				f[j] = 1;
			}	
		}
//		cout<<endl;
	}

	int cnt = 0;
	for (int k = MAX_NUM; k >= 0; k--)//遍历所有的体积结果，为0则不能组合
	{
		if (f[k] == 0) {
			//cout<<k<<" ";
			cnt++;
		}
	}
	cout << cnt;
	return 0;
}