提交时间:2024-07-03 13:29:22
运行 ID: 153479
#include<bits/stdc++.h> using namespace std; int dp[505][505],N,b[505],K; #define dp(i,j) dp[(i)%2][j] 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])));//枚举马厩;枚举前i个马厩放的马匹数;前面的马厩分配了k匹马 printf("%d\n",dp(K,N)); return 0; }