BZOJ4518 征途 [nlogn做法][斜率优化]

我们设x>y，且j由x转移比y转移更优，则
$f[i−1][x]+(sum[j]−sum[x])^2<f[i−1][y]+(sum[j]−sum[y])^2$
化简得
$\frac{f[i−1][x]+sum[x]^2−(f[i−1][y]+sum[y]^2)}{sum[x]−sum[y]}<2∗sum[j]$
那么我们就可以开始斜率优化了……
话说$N^2$的做法真的优秀吗？反正都知道是单调的，不如二分cheak。。。
所以网上居然没有nlogn的题解。。。

#include<bits/stdc++.h>
const int N=200010;
const long long inf=1LL<<25;
using namespace std;
typedef long long ll;
inline ll sqr(ll x){return x*x;}
inline int read(){
    int f=1,x=0;char ch;
    do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');
    do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');
    return f*x;
}
int n,m,lx,rx,a[N],q[N],g[N];
ll s[N],f[N],ans;
inline ll calc1(int x,int y){return f[x]-f[y]+sqr(s[x])-sqr(s[y]);}
inline ll calc2(int x,int y){return s[x]-s[y];}
inline bool check(ll x){
    lx=1;rx=1;q[lx]=0;
    for(int i=1;i<=n;i++){
        while(lx<rx&&calc1(q[lx+1],q[lx])<=2*s[i]*calc2(q[lx+1],q[lx]))lx++;
        f[i]=f[q[lx]]+sqr(s[i]-s[q[lx]])+x;g[i]=g[q[lx]]+1;
        while(lx<rx&&calc1(q[rx],q[rx-1])*calc2(i,q[rx])>=calc1(i,q[rx])*calc2(q[rx],q[rx-1]))rx--;
        q[++rx]=i;
    }
    return g[n]<m;
}
int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=1;i<=n;i++)s[i]=s[i-1]+a[i];
    ll l=0,r=inf;
    while(l<r){
        ll mid=((l+r)>>1)+1;
        if(check(mid))r=mid-1;
        else l=mid,ans=f[n]-l*m;
    }
    printf("%lld\n",ans*m-s[n]*s[n]);
}