hdu2082母函数

找单词

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

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
  
  

想了一段时间没想懂，上网查了发现是种叫母函数的类型的题目，参考了一下http://blog.csdn.net/xiaofei_it/article/details/17042651的模板

代码如下：

#include <iostream>
#include<queue>
#include<vector>

using namespace std;
const int maxnum = 100;
const int maxint = 99999;
const int N = 20000;
const int MAX = 5000;


int a[MAX], b[MAX];

int v[30], n1[30], n2[30];
int P;

int main()
{
    int sum;
    int T;
    int n[30];
    cin >> T;
    P = 50;
    while (T--)
    {
        sum = 0;
        for (int i = 1; i <= 26; i++)
        {
            cin >> n2[i];
            v[i] = i;
        }
       
        a[0] = 1;
        int last = 0;
        for (int i = 1; i<=26; i++)
        {
            int last2 = min(last + n2[i] * v[i], P);
            memset(b, 0, sizeof(int)*(last2 + 1));
            for (int j = 0; j <= n2[i] && j*v[i] <= last2; j++)
                for (int k = 0; k <= last&&k + j*v[i] <= last2; k++)
                    b[k + j*v[i]] += a[k];
            memcpy(a, b, sizeof(int)*(last2 + 1));
            last = last2;
        }
        for (int i = 1; i <= 50; i++)
            if (a[i])
                sum += a[i];
        cout << sum << endl;
    }
    return 0;
}