HDU 2079 选课时间(母函数/生成函数)

本题可以利用01背包+多重背包来解决,也可以利用母函数来解决,博主本着能用数学方法解就不暴力的原则,学习了一下母函数来解决了一下这道题。其中,母函数又叫生成函数,关于它的讲解请参考这篇文章:
http://blog.csdn.net/vsooda/article/details/7975485

关于母函数的实现,可以看我的代码,里面注释很详细,都是我根据自己理解重新注释的

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<set>  
#include<string>  
#include<sstream>
#include<cstdlib>
#define clean(n,m) memset(n,m,sizeof(n))
#define ll long long
using namespace std;
int main()
{
    int T;
    cin >> T;
    while (T--)
    {
        int n, m, sub[10], num[10];
        int c1[41], c2[41];
        //c1存储系数,也就是方案数,c2存储运算中间数据提供给c1更新
        cin >> n >> m;
        for (int i = 1;i <= m;i++)
        {
            cin >> sub[i] >> num[i];
        }
        clean(c1, 0);
        //对c1初始化,由第一个表达式初始化,把从1*sub[1]到num[1]*sub[1]各项(或从1*sub[1]到num[1]*sub[1]各项中直到能达到学分要求的项)的系数都赋成1
        for (int i = 0, t = 0;i <= n&&t <= num[1];i=i+sub[1],t++)
        {
            c1[i] = 1;
        }
        //对c2初始化,全部赋0
        for (int i = 0;i <= n;i++)
        {
            c2[i] = 0;
        }
        //从第二个式子开始,直到遍历完m个式子
        for (int i = 2;i <= m;i++)
        {
            //j就是第i个表达式前面那个式子里的第j个变量,如:(1+x)(1+x^2)(1+x^3),c1[j]先指示的是1x的系数,i=2执行完之后变为(1+x+x^2+x^3)(1+x^3),这时候c1[j]应该指示的是合并后的第一个括号的四个变量的系数
            for (int j = 0;j <= n;j++)
            {
                //k是第j个的指数,k每次增加sub[i]
                for (int k = 0, t = 0;(t <= num[i]) && (j + k <= n);k=k+sub[i], t++)
                {
                    c2[j + k] += c1[j];
                    //这里j+k是保证从sub[i]+j到sub[i]*num[i]+j都加到c2里面,以便后面c1更新数据的时候直接拷贝c2里面的数
                }
            }
            //把c2的值赋给c1,把c2赋成0,因为c2每次是从一个表达式中开始的
            for (int j = 0;j <= n;j++)
            {
                c1[j] = c2[j];
                c2[j] = 0;
            }
        }
        cout << c1[n] << endl;
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值