hdu2082 找单词 (普通型生成函数)

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

分析:

普通型生成函数入门题

假设当前字母为b，则价值为2，若数量为3
则生成函数为:
1x⁰+0x¹+1x²+0x³+1x⁴+0x⁵+1x⁶(系数为0的其实可以省略不写)
ax^b表示组成价值b的方案有a种
即组成价值0一种(一个都不用)，价值2一种(1个2)，价值4一种(两个2)，价值6一种(3个2)
把所有单个价值的生成函数(26种)乘起来就能得到所有价值的生成函数了(多项式相乘)

code1:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm=55;
int a[maxm];//存放答案
int b[maxm];//存放当前字母的生成函数
int c[maxm];//临时存放答案
signed main(){
    int T;
    cin>>T;
    while(T--){
    	memset(a,0,sizeof a);//reset
    	a[0]=1;//不选的时候一种
        for(int k=1;k<=26;k++){
            int num;
            cin>>num;
            if(num==0)continue;
            memset(b,0,sizeof b);//reset
            for(int i=0;i<=num&&i*k<=50;i++){//求当前字母k的生成函数,i*k是指数
                b[i*k]=1;
            }
            for(int i=0;i<=50;i++){//a,b相乘,存入c中
                for(int j=0;j<=50;j++){
                    if(i+j>50)break;
                    c[i+j]+=a[i]*b[j];
                }
            }
            for(int i=0;i<=50;i++){//c复制到a，并清空c
                a[i]=c[i];
                c[i]=0;
            }
        }
        int ans=0;
        for(int i=1;i<=50;i++){//累加答案
            ans+=a[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}

注意到b数组系数最大为1，所以可以优化掉

code2:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm=55;
int a[maxm];
int c[maxm];
signed main(){
    int T;
    cin>>T;
    while(T--){
        memset(a,0,sizeof a);
        a[0]=1;
        for(int k=1;k<=26;k++){
            int num;
            cin>>num;
            if(num==0)continue;
            for(int i=0;i<=num&&i*k<=50;i++){//i*k是当前价值的生成函数指数
                //系数为1
                for(int j=0;j<=50&&i*k+j<=50;j++){
                    c[i*k+j]+=a[j];//其实是+=a[j]*1
                }
            }
            for(int i=0;i<=50;i++){
                a[i]=c[i];
                c[i]=0;
            }
        }
        int ans=0;
        for(int i=1;i<=50;i++){
            ans+=a[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}