HDOJ--2082找单词!！！母函数

原题链接

Problem Description

假设有x1个字母A， x2个字母B,..... x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,..... 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词呢？单词的价值就是组成一个单词的所有字母的价值之和，比如，单词ACM的价值是1+3+14=18，单词HDU的价值是8+4+21=33。(组成的单词与排列顺序无关，比如ACM与CMA认为是同一个单词）。

 

Input

输入首先是一个整数N，代表测试实例的个数。
然后包括N行数据，每行包括26个<=20的整数x1,x2,.....x26.

 

Output

对于每个测试实例，请输出能找到的总价值<=50的单词数,每个实例的输出占一行。

 

Sample Input

  
  

   
   2
1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
9 2 6 2 10 2 2 5 6 1 0 2 7 0 2 2 7 5 10 6 10 2 10 6 1 9
  
  

 

Sample Output

  
  

   
   7
379297
  
  

 

思路：母函数，关于母函数的讲解请移驾 母函数思考与讨论 

代码：

#include<stdio.h>
#include<string.h>
int letter[27];
int c1[51];
int c2[51];
int main()
{
	int N;
	int i,j,k;
	while(scanf("%d",&N)!=EOF)
	{
		while(N--)
		{
			for(i=1;i<=26;i++)
			{
				scanf("%d",&letter[i]);
			}
			memset(c1,0,sizeof(c1));
			memset(c2,0,sizeof(c2));
			c1[0]=1;
			for(i=1;i<=26;i++)
			{
				for(j=0;j<=50;j++)
				{
					for(k=0;k<=letter[i]*i&&j+k<=50;k+=i)
					{
						c2[j+k]+=c1[j];
					}
				}
				for(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);
		}
	}
	return 0;
}