RQNOJ PID273 马棚问题

经典的分组DP

f[i][j]=min{f[k][j-1]+w[k+1][i]}

考虑清楚循环和初始化 

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<climits>
using namespace std;
int w[1000][1000],f[1000][1000],n,k,a[1000],s[1000];
int main()
{
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++){ cin>>a[i]; s[i]=s[i-1]+a[i]; }
for(int i=1;i<n;i++)
for(int j=i+1;j<=n;j++)
w[i][j]=(s[j]-s[i-1])*(j-i+1-s[j]+s[i-1]);
memset(f,63,sizeof(f));
f[0][0]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
for(int k=j-1;k<i;k++)
f[i][j]=min(f[i][j],f[k][j-1]+w[k+1][i]);
cout<<f
[k];
while(1);
return 0;
}