URAL 1167. Bicolored Horses （DP）

题目链接

题意 ：农夫每天都会放马出去，然后晚上把马赶入马厩，于是让马排成一行入马厩，但是不想马走更多的路，所以让前p1匹入第一个马厩，p2匹马入第二个马厩…………但是他不想让他的任何一个马厩空着，所有的马都必须入马厩。有两种颜色的马，如果 i 匹黑马与 j 匹白马同在一个马厩，不愉快系数是 i * j,总系数就是k个系数相加。让总系数最小。

思路 ： dp[i][j] 代表的是前 i 个马厩放 j 匹马的最小不愉快系数值。

//1167
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std ;

int dp[510][510],black[510],white[510] ;

void Init()
{
    for(int i = 0 ; i < 510 ; i++)
    {
        for(int j = 0 ; j < 510 ; j++)
            dp[i][j] = 99999999 ;
        dp[i][i] = 0 ;
    }
    memset(black,0,sizeof(black)) ;
    memset(white,0,sizeof(white)) ;
}
int main()
{
    int N,K,x ;
    while(~scanf("%d %d",&N,&K))
    {
        Init() ;
        for(int i = 1 ; i <= N ; i++)
        {
            scanf("%d",&x) ;
            black[i] = black[i-1] + x ;
            white[i] = white[i-1] + (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-1][k] + (black[j]-black[k])*(white[j]-white[k]) ,dp[i][j]);
                }
            }
        }
        printf("%d
",dp[K][N]) ;
    }
    return 0 ;
}