B - 找单词 母函数

假设有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<stdio.h>
#include<math.h>
#include<stdlib.h>
#include <iostream>
#include<string.h>     
#include<algorithm>
#define ll long long
#define hengheng main ()
const int maxx=1e9;
using namespace std;
long long a[60],b[60];// 用b数组存对应的系数 
 
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        a[0]=1;
        int x;
        for(int i=1;i<=26;i++)
        {
            scanf("%d",&x);
            for(int j=0;j<=50;j++) // 0-50 次方  
            {
                for(int k=0;k<=x&&(j+k*i<=50);k++) //  上一个括号ki 为权值   
                {
                    b[j+k*i]+=a[j];
                   // printf("a:%lld  b:%lld\n ",a[j],b[j+k*i]);
                }
            }
            
            for(int j=0;j<=50;j++)
            {
                a[j]=b[j];   
                b[j]=0;
            }
        }
        long long ans=0;
        for(int i=1;i<=50;i++)
        {
        	 ans+=a[i];
        //	 printf("%lld\n",a[i]);// ai 为其对应的次方数 x的指数 
		}
           
        printf("%lld\n",ans);
    }
}