利用生成函数降低背包问题复杂度

我们知道对于一个完全背包问题,他的算法复杂度大概是O(n^3 ),但是如果利用生成函数大致能降低到O(n^2lgn)
我们来看看咋做,
首先你可以把一个即将向背包里放的一个物品看成是一个多项式,然后呢,每次放一个物品,就把这个物品与前面的那些物品做一次卷积(就是多项式的系数相乘),然后就是找前面的那个系数,对于生成函数这些可以自行知乎(知乎讲的很好),其实多项式乘法这个就有很多东西可以优化,比如可以用FFT,再把多项式乘法的时间降低下来(这个不一定,看是不是做的朴素的多项式乘法了,)然后对于放k个物品这件事,可以用多项式的快速幂方法。
把维数降低到O(lgn),我这只是记个笔记,如果真的想体会的话,可以来做这道题(绝知此事要躬行?

代码:

#include<iostream>
#include<climits>
#include<vector>
#include<cmath>
#include<algorithm>
#define max(a,b) (a>b)?a:b
#define LLONG_MIN -INF
using namespace std;
typedef long long ll;
const int length = 2000;
const ll INF = 0x3f3f3f3f3f3f3f3fll;
vector<ll> v(length);
vector<ll> f(length);

int n;
vector<ll> operator *(vector<ll> &a, vector<ll> &b)
{
	vector<ll> temp(length, LLONG_MIN);
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			temp[(i + j) % n] = max(temp[(i + j) % n], a[i] + b[j]);//这块之前觉得不能这么写是因为他取模了,但是实际上他没有取模
		}
	}
	return temp;
}
bool check(ll k)
{
	vector<ll> base(length, LLONG_MIN);
	base[0] = 0;
	while (k)
	{
		if (k & 1)
		{
			base = v * base;
		}
		v = v * v;
		k = k >> 1;
	}
	for (int i = 0; i < n; i++)
	{
		if (base[i] != f[i])return 0;
	}
	return 1;
}
int main(void)
{
	scanf_s("%d", &n);
	ll max1 = LLONG_MIN;
	for (int i = 1; i <= n; i++)
	{
		ll a;
		scanf_s("%lld", &a);
		max1 = max(max1, a);
		v[i%n] = a;
	}
	ll b=LLONG_MIN;
	ll c;
	for (int i = 0; i < n; i++)
	{
		scanf_s("%lld", &f[i]);
		b = max(b, f[i]);
	}
	
	ll k = b / max1;
	ll m = b % max1;
	if (k<1||k>1e9 || m != 0 || b == 0)
		printf("-1");
	else
	{
		if (check(k))
			printf("%lld", k);
		else
			printf("-1");
	}

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值