包子凑数 (exgcd+dp)

问题描述

　　小明几乎每天早晨都会在一家包子铺吃早餐。他发现这家包子铺有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

思路：dp[i]表示 第i个数是否能凑成，设包子种类为a[i],根据a[i]不断更新dp[i]，即if(dp[j-a[i]])dp[j]=1。   此外还需根据exgcd的原理判断何时取inf

#include<iostream>
#include<cstdio>
#include<map>
#define ll long long
using namespace std;
ll a[1008611],dp[1008611];
int gcd(int a,int b){
	if(b==0)
		return a;
	return gcd(b,a%b);
}
int main(){
	ll n;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	ll d=gcd(a[0],a[1]);
	for(int i=2;i<n;i++){
		d=gcd(a[i],d);
	}
	if(d!=1){
		cout<<"INF"<<endl; 
	}
	else{
	
	dp[0]=1;
	for(int i=0;i<n;i++){
		for(int j=a[i];j<10000;j++){
			if(dp[j-a[i]])dp[j]=1;
		}
	}
	ll  ans=0;
	for(int i=1;i<10000;i++){
		if(!dp[i])ans++;
	}
	cout<<ans<<endl;
}
	return 0;
}