hdu2082——找单词

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/westbrook1998/article/details/82023699

假设有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

用母函数解法 直接套模板 然后26个多项式相乘这样子

代码：

#include <cstdio>
#include <algorithm>
#include <cstring>
#define _clr(x,a) memset(x,a,sizeof(x));
using namespace std;
const int N=1000000;
long long tmp[N];
long long ans[N];
int main(void){
    int t;
    scanf("%d",&t);
    while(t--){
        _clr(tmp,0);
        _clr(ans,0);
        ans[0]=1;
        int x;
        for(int i=1;i<=26;i++){
            //各个字母的系数
            scanf("%d",&x);
            for(int j=0;j<=50;j++){
                for(int k=0;k<=x && j+k*i<=50;k++){
                    tmp[j+k*i]+=ans[j];
                }
            }
            for(int j=0;j<=50;j++){
                ans[j]=tmp[j];
                tmp[j]=0;
            }
        }
        long long res=0;
        //系数为0不能加进去
        for(int i=1;i<=50;i++){
            res+=ans[i];
        }
        printf("%lld\n",res);
    }
    return 0;
}