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
分析:
普通型生成函数入门题
假设当前字母为b,则价值为2,若数量为3
则生成函数为:
1x0+0x1+1x2+0x3+1x4+0x5+1x6(系数为0的其实可以省略不写)
axb表示组成价值b的方案有a种
即组成价值0一种(一个都不用),价值2一种(1个2),价值4一种(两个2),价值6一种(3个2)
把所有单个价值的生成函数(26种)乘起来就能得到所有价值的生成函数了(多项式相乘)
code1:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm=55;
int a[maxm];//存放答案
int b[maxm];//存放当前字母的生成函数
int c[maxm];//临时存放答案
signed main(){
int T;
cin>>T;
while(T--){
memset(a,0,sizeof a);//reset
a[0]=1;//不选的时候一种
for(int k=1;k<=26;k++){
int num;
cin>>num;
if(num==0)continue;
memset(b,0,sizeof b);//reset
for(int i=0;i<=num&&i*k<=50;i++){//求当前字母k的生成函数,i*k是指数
b[i*k]=1;
}
for(int i=0;i<=50;i++){//a,b相乘,存入c中
for(int j=0;j<=50;j++){
if(i+j>50)break;
c[i+j]+=a[i]*b[j];
}
}
for(int i=0;i<=50;i++){//c复制到a,并清空c
a[i]=c[i];
c[i]=0;
}
}
int ans=0;
for(int i=1;i<=50;i++){//累加答案
ans+=a[i];
}
cout<<ans<<endl;
}
return 0;
}
注意到b数组系数最大为1,所以可以优化掉
code2:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm=55;
int a[maxm];
int c[maxm];
signed main(){
int T;
cin>>T;
while(T--){
memset(a,0,sizeof a);
a[0]=1;
for(int k=1;k<=26;k++){
int num;
cin>>num;
if(num==0)continue;
for(int i=0;i<=num&&i*k<=50;i++){//i*k是当前价值的生成函数指数
//系数为1
for(int j=0;j<=50&&i*k+j<=50;j++){
c[i*k+j]+=a[j];//其实是+=a[j]*1
}
}
for(int i=0;i<=50;i++){
a[i]=c[i];
c[i]=0;
}
}
int ans=0;
for(int i=1;i<=50;i++){
ans+=a[i];
}
cout<<ans<<endl;
}
return 0;
}