HDOJ2082母函数

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

本题最容易想到的是暴力循环,但是这种方法计算规模及其庞大,假设所有字母有一个,那么就有2的26次方种情况,更不用说每个单词有20个。

本题一种比较好的办法是母函数。
所谓母函数,概括性的描述:把组合问题的加法法则和幂级数的的乘幂的相加对应起来,把离散数列间的相互结合关系对应成为幂级数间的运算关系,最后由幂级数形式来确定离散数列的构造。

从那个经典的砝码的例子讲起。
题目:有1克、2克、3克、4克的砝码各一 枚,能称出哪几种重量?每种重量各有几种可能方案?
母函数法将砝码总克数作为指数
如果只用1克砝码,则表示为x^0 + x^1,分别代表用0个和用1个的情况
如果只用2克砝码,则表示为x^0 + x^2,分别代表用0个和用1个的情况
如果只用3克砝码,则表示为x^0 + x^3,分别代表用0个和用1个的情况
如果只用4克砝码,则表示为x^0 + x^4,分别代表用0个和用1个的情况
那么如何组合起来呢?指数代表克数,组合时克数相加,那么指数相加就对应表达式相乘;
(x^0 +x^1)* (x^0 +x^2) * (x^0 +x^3)* (x^0 +x^4)

=x^0 + x^1 + x^2 + 2x^3 + 2x^4 + 2x^5 + 2x^6 + 2x^7 + x^8+ x^9 + x^10

每项的指数代表总克数,系数代表组合数,比如4克有1个4克或者一个1克和两个三克两种。
联系到本题,每个字母也可以看做一种砝码,字母个数即为砝码个数,我们只需求出最终表达式中指数小于等于50的项的系数和即可。

#include <iostream>
#include<cstdio>
#include<cstring>
#include <iomanip>
# include <cmath>
using namespace std;

int main()
{ 
 int i, j, k,n,num[27];
 long long t[51], s[51],sum;//s[i]存储当前表达式中指数为i的项的系数,t为计算时的临时变量。
 while (cin >> n)
 {
  while (n)
  {
   for (i = 1; i <= 26; i++)
    scanf_s ("%d",&num[i]);
   memset(t, 0, sizeof(t));
   memset(s, 0, sizeof(s));
   for (i = 0; i <= num[1]; i++)
    s[i] = 1;//只用A,则0到num[1]都只有一种情况。
   for (i = 2; i <= 26; i++)//考虑使用i对应字母
   {
    for (j = 0; j <= 50; j++)//计算出的对应指数为j
     for (k = 0; j + k <= 50 && k / i <= num[i]; k += i)//加入总值为k的i对应字母,因而k每次增加i
      t[j + k] += s[j];//求出j+k指数的增量,k为i字母的总值,只有一种情况,j值有s[j]种组合,所以这里实际上是+=s[j]*1
    for (j = 0; j <= 50; j++)
    {
     s[j] = t[j];//合并同类项
     t[j] = 0;//临时变量清零
    }
   }
   sum = 0;
   for (i = 1; i <= 50; i++)
    sum += s[i];
   cout << sum << endl;
   n--;
  }
 }
 return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值