hdu 2082

母函数计数算法

最新推荐文章于 2019-04-19 09:40:53 发布

原创最新推荐文章于 2019-04-19 09:40:53 发布 · 279 阅读

0 ·

CC 4.0 BY-SA版权

母函数专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用母函数进行计数的算法实现方法。通过输入数组a[1...26]，对于每个非零元素，算法计算出所有可能的组合，并记录下组合的个数。最终输出的是不超过50的所有可能的组合数量。

这里有解释

母函数

代码如下

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int c1[51],c2[51];
int a[30];
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int i;
		memset(c1,0,sizeof(c1));
		memset(c2,0,sizeof(c2));
		for(i=1;i<=26;i++)
			scanf("%d",&a[i]);
		for(i=1;i<=26;i++){
			if(a[i]!=0){
				for(int j=0;j<=a[i];j++){
					if(i*j>50) break;
					c1[i*j]=1;
				}
				break;
			}
		}
		for(i=i+1;i<=26;i++){
			if(a[i]!=0){
				for(int j=0;j<=50;j++){
					if(c1[j]!=0){
						for(int k=0;k<=a[i];k++)
						{
							if(j+k*i>50) break;
							c2[j+k*i]+=c1[j];
						}
					}
				}
				for(int j=0;j<=50;j++)
				{
					c1[j]=c2[j];
					c2[j]=0;
				}
			}
		}
		int sum=0;
		for(i=1;i<=50;i++)
			sum+=c1[i];
		printf("%d\n",sum);
	}
}