斜率优化第二题。。
题意自不用说,传送门
还是一样的套路啊,因为决策点过多,所以我们不可能枚举所有决策点,然后就要对决策点进行筛选,只有优于已经选择了的决策点的决策点才能被选择。
问题是怎么才算优于?就要斜率优化。
如果暴力枚举,n^2,明显可以列出DP式子。
dp[i]=min(dp[j]+(sum[i]−sum[j]+i−j−1−l)2)
这里是选择从j+1到i.
那么我们进行化简,设c=1+l,f[i]=sum[i]+i:
dp[i]=min(dp[j]+(f[i]−f[j]−c))(c=1+l)
然后我们就可以斜率优化了。
具体单调性的证明请移步hzwer
假设有决策点j,k,k比j优秀。
dp[k]+(f[i]−f[k]−c)2<=dp[j]+(f[i]−f[j]−c)2
。。。由于步骤过长我就不全写了,最后化简得出:
(dp[k]+(f[k]+c)2−dp[j]−(f[j]+c)2)/2∗(f[k]−f[j])<=f[i]
按照这个单调队列就好了。
如果卡精度的话还要移项,但是不卡就算了。
写个函数转移就好.
#include<cstdio>
#include<algorithm>
#include<cstring>
#define sqr(x) ((x)*(x))
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=2e5+5;
int n,m,a[N],q[N];
ll f[N],sum[N],C;
inline double pd(int j,int k)
{
return (f[k]-f[j]+sqr(sum[k]+C)-sqr(sum[j]+C))/(2.0*(sum[k]-sum[j]));
}
int main()
{
scanf("%d%d",&n,&m);
C=m+1;
fo(i,1,n)scanf("%d",&a[i]);
fo(i,1,n)sum[i]=sum[i-1]+a[i];
fo(i,1,n)sum[i]+=i;
int t=1,w=1;
fo(i,1,n)
{
while (t<w&&pd(q[t],q[t+1])<=sum[i])t++;
int j=q[t];
f[i]=f[j]+sqr(sum[i]-sum[j]-C);
while (t<w&&pd(q[w],i)<pd(q[w-1],q[w]))w--;
q[++w]=i;
}
printf("%lld\n",f[n]);
}