|
| 1 | +#include <iostream> |
| 2 | +#include <string> |
| 3 | +#include <vector> |
| 4 | +#include <algorithm> |
| 5 | +#include <queue> |
| 6 | +#include <list> |
| 7 | +#include <stack> |
| 8 | +#include <map> |
| 9 | +#include <set> |
| 10 | +using namespace std; |
| 11 | + |
| 12 | +#define DEBUG |
| 13 | +#define REP(i,a) for(i=0;i<a;i++) |
| 14 | +#define FOR(i,a,b) for(i=a;i<b;i++) |
| 15 | +#define VE vector<int> |
| 16 | +#define SZ size() |
| 17 | +#define PB push_back |
| 18 | +const int MAXN = 502; |
| 19 | +int N, K, h[MAXN]; |
| 20 | + |
| 21 | +//int dp[MAXN][MAXN]; |
| 22 | +int dp[MAXN]; |
| 23 | +int zero[MAXN], one[MAXN]; |
| 24 | +const int INF = 99999999; |
| 25 | +void pre(){ |
| 26 | + //zero[1] = (h[0]==0?1:0); |
| 27 | + //one[1] = (h[0]==1?1:0); |
| 28 | + for(int i=1; i<=N; i++){ |
| 29 | + if(h[i-1] == 1){ |
| 30 | + one[i] += one[i-1] + 1; |
| 31 | + zero[i] = zero[i-1]; |
| 32 | + }else{ |
| 33 | + zero[i] += zero[i-1]+1; |
| 34 | + one[i] = one[i-1]; |
| 35 | + } |
| 36 | + //cout<<"zero["<<i<<"] = "<<zero[i]<<endl; |
| 37 | + //cout<<"one["<<i<<"] = "<<one[i]<<endl; |
| 38 | + } |
| 39 | +} |
| 40 | + |
| 41 | + |
| 42 | +void work(){ |
| 43 | + pre(); |
| 44 | + for(int i=0; i<MAXN; i++)for(int j=0; j<MAXN; j++){ |
| 45 | + //dp[i][j ] = INF; |
| 46 | + dp[i] = INF; |
| 47 | + } |
| 48 | + |
| 49 | + //dp[0][0] = 0; |
| 50 | + dp[0] = 0; |
| 51 | + |
| 52 | + for(int k=0; k<K; k++){ |
| 53 | + for(int i=1; i<=N; i++){ |
| 54 | + for(int j=0; j<i; j++){ |
| 55 | + int n_one = one[i] - one[j]; |
| 56 | + int n_zero = zero[i] - zero[j]; |
| 57 | + int v = n_one * n_zero; |
| 58 | + cout<<i<<" "<<j<<" v = "<<n_one<<"*"<<n_zero<<" = "<<v<<endl; |
| 59 | + //dp[i][k+1] = min(dp[i][k+1], dp[j][k]+v); |
| 60 | + dp[i] = min(dp[i], dp[j] + v); |
| 61 | + //cout<<"dp["<<i<<"]["<<k+1<<"] = "<<dp[i][k+1]<<endl; |
| 62 | + cout<<"dp["<<i<<"] = "<<dp[i]<<endl; |
| 63 | + } |
| 64 | + } |
| 65 | + } |
| 66 | + |
| 67 | + //cout<<dp[N][K]<<endl; |
| 68 | + for(int i=0; i<=N; i++){ |
| 69 | + cout<<dp[i]<<" "; |
| 70 | + }cout<<endl; |
| 71 | + |
| 72 | + cout<<dp[N]<<endl; |
| 73 | + |
| 74 | +} |
| 75 | + |
| 76 | +int main(){ |
| 77 | + cin>>N>>K; |
| 78 | + for(int i=0; i<N; i++){ |
| 79 | + cin>>h[i]; |
| 80 | + } |
| 81 | + work(); |
| 82 | + return 0; |
| 83 | + |
| 84 | +} |
| 85 | + |
0 commit comments