题目描述
题解
我们可以得到裸的dp:
f[i]=f[j]+(sum[i]−sum[j])2∗a+(sum[i]−sum[j])∗b+c
然后展开。
f[i]=f[j]+a∗sum[i]2−2∗a∗sum[i]∗sum[j]+a∗sum[j]2+b∗sum[i]−b∗sum[j]+c
然后整理,准备斜率优化,最后得到
f[j]+a∗sum[j]2−b∗sum[j]=2∗a∗sum[i]∗sum[j]+f[i]−a∗sum[i]2−b∗sum[i]−c
y=f[j]+a∗sum[j]2−b∗sum[j]
k=2∗a∗sum[i]
x=sum[j]
然后就可以愉快的斜率优化了。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
typedef long long ll;
const int maxn = 100000+10;
using namespace std;
int n,l,r;
ll a,b,c,q[maxn],f[maxn],s[maxn];
inline int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
inline double getk(int x,int y)
{
return (double)((f[x]+a*s[x]*s[x]-b*s[x])-(f[y]+a*s[y]*s[y]-b*s[y]))/(double)(s[x]-s[y]);
}
int main()
{
n=read();a=read();b=read();c=read();
F(i,1,n) s[i]=s[i-1]+read();
l=r=1;q[1]=0;
F(i,1,n)
{
while (l<r&&getk(q[l],q[l+1])>a*s[i]*2) l++;
f[i]=f[q[l]]+a*s[q[l]]*s[q[l]]-b*s[q[l]]-a*s[i]*s[q[l]]*2+a*s[i]*s[i]+b*s[i]+c;
while (l<r&&getk(q[r-1],q[r])<getk(q[r],i)) r--;
q[++r]=i;
}
printf("%lld\n",f[n]);
return 0;
}