当前位置: 代码迷 >> 综合 >> 【斜率优化dp】P3628 [APIO2010]特别行动队
  详细解决方案

【斜率优化dp】P3628 [APIO2010]特别行动队

热度:55   发布时间:2024-02-06 09:10:30.0

首先定义f[i]为前i个士兵组成的最大战斗力

直接计算的方式

    for(int i=1;i<=n;++i)for(int j=0;j<i;++j)f[i]=max(f[i],f[j]+calc(c[i]-c[j]));

其中calc为计算二次函数的值  c[i]为预处理的前缀和

这样只能得到50分,我们还需要继续进行斜率优化

设j的方案比k更优,有

整理得

 

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int n,x[maxn];
long long f[maxn],s[maxn],c[maxn];
int A,B,C;
long long calc(long long x)
{return x*x*A+x*B+C;
}
double count(long long j , long long k)
{return (f[j]-B*c[j]+A*c[j]*c[j]-f[k]-A*c[k]*c[k]+B*c[k])/(2.0*A*(c[j]-c[k]));
}
int main()
{freopen("a.in","r",stdin);	freopen("a.out","w",stdout);scanf("%d%d%d%d",&n,&A,&B,&C);for(int i=1;i<=n;i++) scanf("%d",&x[i]);for(int i=1;i<=n;i++) c[i]=c[i-1]+x[i];for(int i=1;i<=n;i++) f[i]=-1e18;int h=0,t=0;for(int i=1;i<=n;i++){while(h<t && count(s[h],s[h+1])<=c[i]*1.0) h++;f[i]=f[s[h]]+calc(c[i]-c[s[h]]);while(h<t && count(s[t-1],s[t])>=count(s[t],i)) t--;s[++t]=i;}printf("%lld\n",f[n]);return 0;
}