P1269 马棚

P1269 马棚
2017年8月4日
动态规划

---

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

int N, K;//把n匹马放入k个马棚;
#define maxx 610
int Kind[maxx];//1意味着马是黑色的，0意味着马是白色的;
int T[maxx][maxx];//从第i匹马到第k匹马的不愉快系数;
int F[maxx][maxx];//动态规划：不愉快系数;

void Data()
{
    for(int i = 1; i <= N; i++)
        for(int k = i; k <= N; k++){
            int t1 = 0, t0 = 0;
            for(int j = i; j <= k; j++)
                if(Kind[j] == 1)    t1++;
                else    t0++;
            T[i][k] = t1 * t0;
        }
    return;
}

void DP()
{
    memset(F, 10, sizeof(F));
    for(int i = 1; i <= N; i++)
        F[1][i] = T[1][i];
    for(int i = 2; i <= K; i++)
        for(int k = i; k <= N; k++)
            for(int j = 1; j <= k - 1; j++)
                F[i][k] = min(F[i][k], F[i - 1][j] + T[j + 1][k]);
}

int main()
{
    cin >> N >> K;
    for(int i = 1; i <= N; i++)    cin >> Kind[i];
    Data();
    DP();
    cout << F[K][N] << endl;
    return 0;
}