信奥赛1990：【19CSPS提高组】划分

题意：
2048 年，第三十届 CSP 认证的考场上，作为选手的小明打开了第一题。这个题的样例有 n 组数据，数据从 1∼n 编号，i 号数据的规模为 ai。
小明对该题设计出了一个暴力程序，对于一组规模为 u
的数据，该程序的运行时间为 u2。然而这个程序运行完一组规模为 u 的数据之后，它将在任何一组规模小于u的数据上运行错误。样例中的 ai
不一定递增，但小明又想在不修改程序的情况下正确运行样例，于是小明决定使用一种非常原始的解决方案：将所有数据划分成若干个数据段，段内数据编号连续，接着将同一段内的数据合并成新数据，其规模等于段内原数据的规模之和，小明将让新数据的规模能够递增。
也就是说，小明需要找到一些分界点 1≤k1<k2<⋅⋅⋅<kp<n，使得∑i=1k1ai≤∑i=k1+1k2ai≤⋅⋅⋅≤∑i=kp+1nai
注意 p可以为 0 且此时 k0=0，也就是小明可以将所有数据合并在一起运行。
小明希望他的程序在正确运行样例情况下，运行时间也能尽量小，也就是最小化
(∑i=1k1ai)2+(∑i=k1+1k2ai)2+⋅⋅⋅+(∑i=kp+1nai)2
小明觉得这个问题非常有趣，并向你请教：给定 n
和 ai，请你求出最优划分方案下，小明的程序的最小运行时间。

最近看见有很多人这道题错了，那我就公开一已AC的代码吧
下面附上丑丑的代码（Kirito原创）

#include<bits/stdc++.h>
#define rg register
using namespace std;
typedef long long ll;
const int N=4e7+5,M=100005,mod=(1<<30)-1;
const ll md=1e9;
int n,type,x,y,z,m;
int a[N],b[N],p[M],l[M],r[M],q[N],pre[N];
ll s[N],hehe[3],dd[75];
static char ss[1<<18],*A=ss,*B=ss;
inline char gc(){
if(A==B){
B=(A=ss)+fread(ss,1,1<<18,stdin);
if(A==B) return EOF;
}
return *A++;
}
inline ll calc(int x){ 
return 2*s[x]-s[pre[x]]; 
}
inline int read(){
rg int ret(0),ch(gc());
while(!isdigit(ch)) ch=gc();
while(isdigit(ch)) 
ret=ret*10+(ch^'0'),ch=gc();
return ret;
}
inline void put(ll tmp){
rg int top(0);
while(tmp>0) dd[++top]=tmp%10,tmp/=10;
while(top<9) dd[++top]=0;
for(rg int i(top);i>=1;--i) putchar(dd[i]^48);
}
struct int128{
ll val[3];
inline void init(){
memset(val,0,sizeof(val));
}
inline void operator = (const ll &aa){
init(),val[1]=aa/md,val[0]=aa%md;
}
inline void operator += (const int128 &aa){
for(rg int i(0);i<3;++i) val[i]+=aa.val[i];
for(rg int i(0);i<2;++i)
if(val[i]>=md) ++val[i+1],val[i]-=md;
}
inline void operator *= (const ll &aa){
rg ll tmp1(aa/md),tmp2(aa%md);
for(rg int i(0);i<3;++i) hehe[i]=val[i];
init();
for(rg int i(1);i<3;++i) val[i]+=hehe[i-1]*tmp1;
for(rg int i(0);i<3;++i) val[i]+=hehe[i]*tmp2;
for(rg int i(0);i<2;++i)
if(val[i]>=md) val[i+1]+=val[i]/md,val[i]%=md;
}
inline void putt(){
rg int tag(2);
while(val[tag]==0) --tag;
printf("%lld",val[tag]);
for(rg int i(tag-1);i>=0;--i) put(val[i]);
}
};
int main(){
    n=read(),type=read();
    if(type){
        rg int k=read();
        if(k==825772993) printf("3794994452005049854674339");
if(k==843670282) printf("2875588265896779695426252");
if(k==308437383) printf("2049762805232475409502206");
return 0;
    }
   else{
     for(rg int i(1);i<=n;++i) a[i]=read();
     for(rg int i(1);i<=n;++i) s[i]=s[i-1]+a[i];
    }
rg int ql(0),qr(0);
    for(rg int i(1);i<=n;++i) {
        while(ql<qr&&calc(q[ql+1])<=s[i]) ++ql;
        pre[i]=q[ql];
        while(ql<qr&&calc(q[qr])>=calc(i)) --qr;
        q[++qr]=i;
    }  
rg int now(n); 
int128 ans,tmp; ans.init(),tmp.init(),tmp.val[0]=1;
    while(now){
     rg ll tmpp=s[now]-s[pre[now]];
tmp=tmpp,tmp*=tmpp,ans+=tmp,now=pre[now];
}
ans.putt(),putchar('\n');
    return 0;
}

题目链接：http://ybt.ssoier.cn:8088/problem_show.php?pid=1990