杭电2082

                    找单词

http://acm.hdu.edu.cn/showproblem.php?pid=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 <iostream>
# include <cstring>
using namespace std;

int c1[100009],c2[100009];
int main(){

    int T,n,i,j,k;
    int a[30];


    while(cin>>T){

        while(T--){

            for(i=1;i<=26;i++){
                cin>>a[i];
            }

            memset(c1,0,sizeof(c1));
            memset(c2,0,sizeof(c2));    

            for(i=0;i<=a[1];i++){
                c1[i] = 1;
            }

            int len = a[1]*1;
            for(i=2;i<=26;i++){
                for(j=0;j<=len;j++){
                    for(k=0;k<=a[i]*i;k+=i){
                        c2[j+k] +=c1[j]; 
                    }
                }

                len += a[i]*i;
                for(k=0;k<=len;k++){
                    c1[k] = c2[k];
                    c2[k] = 0; 
                }
            }

            int  cnt = 0;
            /* 错误的写法: for(k=1;k<=50;k++){ if(c1[k]<=50){ cnt+=c1[k]; } } */

            //正确的写法: 
            for(k=1;k<=50;k++){
                cnt+=c1[k];
            }
            /* 原因: 记住在a[n]=b 等价于bx^n b代表在n次幂下的种类 */
            cout<<cnt<<endl; 
        }



    }


    return 0;
}

你可能感兴趣的:(杭电,母函数)