比赛 2024国庆练习3 评测结果 AAAAAAAAAAAAAAAATTTTTTWWW
题目名称 划分 最终得分 64
用户昵称 郑霁桓 运行时间 30.223 s
代码语言 C++ 内存使用 5.52 MiB
提交时间 2024-10-06 17:47:24
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
long long n,p,a[500005],b[500005];
long long l[500005],f[500005];
int main(){
    freopen("2019partition.in","r",stdin);
    freopen("2019partition.out","w",stdout);
    cin>>n>>p;
    if(p){ cout<<0; return 0;}
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=b[i-1]+a[i];
    }
    f[1]=a[1]*a[1],l[1]=a[1];
    for(int i=2;i<=n;i++){
        for(int j=0;j<i;j++){
            if(b[i]-b[j]>=l[j]){
                f[i]=f[j]+(b[i]-b[j])*(b[i]-b[j]);
                l[i]=b[i]-b[j];
            }
        }
    }
    cout<<f[n];
	return 0;
}