我们知道对于一个完全背包问题,他的算法复杂度大概是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");
}
}