ural Bicolored Horses(二维dp)

http://acm.timus.ru/problem.aspx?space=1&num=1167


有n个马,黑白两种,依次放入k个马厩,将x匹马放在一个马厩的不快乐值为黑马数目*白马数目。问最后的不快乐值最小是多少?


设dp[i][j]表示前i个马厩放了j匹马的最小不快乐值,那么dp[i][j] = min(dp[i-1][g]+tmp[g+1][j])。

其中tmp是预处理的i到j匹马的不快乐值。


#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
#define PP pair<LL,LL>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 100010;
const int mod = 1000000007;
int n,k;
int a[510];
int tmp[510][510];
int t[2];
int dp[510][510];

int main()
{
    while(~scanf("%d %d",&n,&k))
    {
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i = 1; i < n; i++)
        {
            t[0] = t[1] = 0;
            t[a[i]] += 1;
            tmp[i][i] = 0;
            for(int j = i+1; j <= n; j++)
            {
                t[a[j]] += 1;
                tmp[i][j] = t[0]*t[1];

            }
        }
        memset(dp,INF,sizeof(dp));
        for(int i = 1; i <= n; i++)
		{
			dp[1][i] = tmp[1][i];
		}
		for(int i = 2; i <= k; i++)
		{
			for(int j = i; j <= n; j++)
			{
				int Min = INF,pos;
				for(int g = i-1; g <= j; g++)
				{
					if(g+1 <= j)
						Min = min(Min,dp[i-1][g]+tmp[g+1][j]); //检查好久,错把这个放在循环外面了。
				}
				dp[i][j] = Min;
			}
		}
        printf("%d\n",dp[k][n]);
    }
    return 0;
}


你可能感兴趣的:(dp)