hdu 2082 找单词（母函数）

找单词

Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 8128 Accepted Submission(s): 5699

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

---

题意：还是说一下，每个字母有自己的价值，1~26，给出每个字母的个数，求出组成价值50以内的单词个数。

---

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 31
int a[N], b[N], nm1[2*N], nm2[2*N];
int main()
{
    freopen( "in.txt", "r", stdin );
    int i, j, k, n;
    for( i = 0 ; i <= N ; i ++ ){
        b[i] = i+1;/// != i;
    }
    scanf( "%d", &n );
    while( n -- ){
        memset( nm1, 0, sizeof( nm1 ));
        memset( nm2, 0, sizeof( nm2 ));
        for( i = 0 ; i < 26 ; i ++ ){
            scanf( "%d", &a[i] );
        }
        for( i = 0 ; i <= a[0] ; i ++ ){
            nm1[i] = 1;
        }
        for( i = 1 ; i < 26 ; i ++ ){
            for( j = 0 ; j <= 50 ; j ++ ){
                if( nm1[j] )
                for( k = 0 ; k <= a[i]*b[i] && k+j <= 50 ; k += b[i] ){
                ///k+j<=50为了减少不必要的运算，还会超数组。
                    nm2[k+j] += nm1[j];
                }
            }
            memcpy( nm1, nm2, sizeof( nm1 ));
            memset( nm2 , 0, sizeof( nm2 ));
        }
        int sum = 0;
        for( i = 1 ; i <= 50 ; i ++ ){
            sum += nm1[i];
            //cout<<nm1[i]<<" "<<sum<<endl;
        }
        cout<<sum<<endl;
    }
    return 0;
}