人类智慧过了

1234567891011121314151617181920212223242526#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m,a[3005],f[3005][3005];
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;
	int sum=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		a[i]+=a[i-1];
	}
	memset(f,0x3f,sizeof(f));
	f[0][0]=0;
	for(int i=1;i<=n;i++){
		int p=lower_bound(a+0,a+i,a[i]-a[n]/m)-a;
		for(int k=1;k<=m;k++){
			for(int j=max(1ll,p-2);j<=min(i,p+2);j++){
				f[i][k]=min(f[i][k],f[j-1][k-1]+(a[i]-a[j-1])*(a[i]-a[j-1]));
			}
		}
	}
	cout<<-a[n]*a[n]+m*f[n][m];
	return 0;
}