老规矩,方程什么都在注释里面。
呃,其实写斜率优化不妨像我这么写,先把会TLE的裸dp交一遍,T了以后就把方程去掉(long long)这种类型转化神马的粘到注释里面,然后j放一边,i放一边,都有的放到i的一边,然后列出yxkb,然后最大上凸,最小下凹,简单分析,不对再改,然后来俩点判断判断叉积神马的,然后一遍ac。
嘲讽一下:某同学POJ一道题调了一天至今还没过。。代码风格啊。。。
不会斜率优化的往我博客分类里其它博客翻。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1001000
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3fll
/*
2*f[i]=2*f[j]+i*i+j*j-2*i*j-i+j+2*a[i]
2*f[j]+j*j+j=2*i*j+2*f[i]-i*i+i-2*a[i]
y=2*f[j]+j*j+j
x=j
k=2*i
b=2*f[i]-i*i+i-2*a[i]
*/
using namespace std;
int n,a[N];
struct Point
{
long long x,y;
int id;
Point(long long _x,long long _y,int _id):x(_x),y(_y),id(_id){}
Point(){}
}now,q[N];
int l,r;
long long f[N];
inline long long xmul(Point i,Point j,Point k){return (i.y-k.y)*(j.x-k.x)-(j.y-k.y)*(i.x-k.x);}
int main()
{
// freopen("test.in","r",stdin);
// freopen("my.out","w",stdout);
int i,j;
scanf("%d",&n);
memset(f,0x3f,sizeof(f)),f[0]=0;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
now=Point(i-1,2ll*f[i-1]+(long long)i*i-i+2,i-1);
while(l<r&&xmul(now,q[r],q[r-1])<=0)r--; //斜(i,k) 斜(j,k) 弹
q[++r]=now;
long long K=2*i;
while(l<r&&q[l+1].y-q[l].y<=(q[l+1].x-q[l].x)*K)l++;
j=q[l].id;
f[i]=f[j]+((long long)i*i+(long long)j*j-2ll*i*j-i+j)/2+a[i];
}
// for(i=1;i<=n;i++)printf("%d\n",f[i]);
cout<<f[n]<<endl;
return 0;
}