HDU 2082 找单词(母函数基础)

H - 找单词

Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Submit  Status

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<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

const int N=50;

int c1[N+10],c2[N+10],num[30];

int main(){

    //freopen("input.txt","r",stdin);

    int t;
    scanf("%d",&t);
    while(t--){
        memset(c1,0,sizeof(c1));    //c1[ ]保存当前得到的多项式各项系数
        memset(c2,0,sizeof(c2));    //c2[ ]保存每次计算时的临时结果
        for(int i=1;i<=26;i++)
            scanf("%d",&num[i]);
        c1[0]=1;                //相当于用X^0去乘以后面的多项式 
        for(int i=1;i<=26;i++){     //要乘以26个多项式
            for(int j=0;j<=N;j++)   //c1的各项的指数
                for(int k=0;k<=num[i] && j+k*i<=N;k++)  //k*i表示被乘多项式各项的指数,(X^0*i + X^1*i + X^2*i + ……)
                    c2[j+k*i]+=c1[j];       //指数相加得j+k*i,加多少只取决于c1[j]的系数，因为被乘多项式的各项系数均为1
            for(int j=0;j<=N;j++){
                c1[j]=c2[j];
                c2[j]=0;
            }
        }
        int ans=0;
        for(int i=1;i<=N;i++)
            ans+=c1[i];
        printf("%d\n",ans);
    }
    return 0;
}