提交时间:2024-05-25 16:41:05

运行 ID: 149136

#include <bits/stdc++.h> using namespace std; #define dp(i,j) dp[(i)%2][j] int N,K,dp[501][501],b[501]; int main(){ cin>>N>>K; for(int i=0; i<2; i++) for(int j=i+1; j<501; j++) dp[i][j]=1<<30; for(int i=1,x; i<=N; i++){ scanf("%d",&x); b[i]=b[i-1]+x; } for(int i=1; i<=K; i++) for(int j=i; j<=N-(K-i); j++) for(int k=i-1; k<j; k++) dp(i,j)=min(dp(i,j),dp(i-1,k)+(b[j]-b[k])*(j-k-(b[j]-b[k]))); printf("%d\n",dp(K,N)); }