Holding Bin-Laden Captive!（HDU-1085）

最新推荐文章于 2024-10-02 22:45:05 发布

cjmHK

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量183

点赞数

分类专栏：生成函数文章标签：算法

本文链接：https://blog.csdn.net/qq_40634175/article/details/89440469

版权

生成函数专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题意

给出三个数 $1, 2, 5$ ，其数量分别为 $a_{1}, a_{2}, a_{3}$ ，问这三个数不能凑出的最小数。
链接：link。

思路

对于这类组合计数问题，可以用生成函数解决，将计数问题转化为多项式乘法问题。相关知识请参考《组合数学》。

时间复杂度分析

多项式相乘与多项式最高次数有关，而多项式最高次数为 $\mathcal{O}(a_{1}+2a_{2}+5a_{3})$ 。所以时间复杂度为 $\mathcal{O}((a_{1}+2a_{2}+5a_{3})^{2})$ 。

实现

#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int N=8005;
int cnt[N],tmp[N];
int main(){
    int a,b,c;
    while(cin>>a>>b>>c&&(a||b||c)){
        memset(tmp,0,sizeof(tmp));
        for(int i=0;i<=a;i++){
            for(int j=0;j<=2*b;j+=2){
                tmp[i+j]+=1;
            }
        }
        for(int i=0;i<=a+2*b;i++) cnt[i]=tmp[i];
        memset(tmp,0,sizeof(tmp));
        for(int i=0;i<=a+2*b;i++){
            for(int j=0;j<=5*c;j+=5){
                tmp[i+j]+=cnt[i];
            }
        }
        for(int i=0;i<=a+2*b+5*c+1;i++){
            if(!tmp[i]){
                cout<<i<<endl;
                break;
            }
        }
    }
    return 0;
}