关闭

【动态规划】Codeforces 711C Coloring Trees

标签: 动态规划Codeforces题解
494人阅读 评论(0) 收藏 举报
分类:

题目链接:

  http://codeforces.com/problemset/problem/711/C

题目大意:

  给N棵树,M种颜色,已经有颜色的不能涂色,没颜色为0,可以涂色,每棵树I涂成颜色J花费PIJ。求分成K个颜色段(1112221为3个颜色段)的最小花费。无解输出-1.

题目思路:

  【动态规划】

  f[i][j][k]表示前i个树分成j段,最后一个颜色是k的花费。

  根据当前这棵树是否可以涂色,上一棵树是否可以涂色转移。

  直接枚举这个树和上棵树的颜色,N4居然没T。 




//
//by coolxxx
//#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<iomanip>
#include<map>
#include<stack>
#include<queue>
#include<set>
#include<bitset>
#include<memory.h>
#include<time.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
//#include<stdbool.h>
#include<math.h>
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))
#define abs(a) ((a)>0?(a):(-(a)))
#define lowbit(a) (a&(-a))
#define sqr(a) ((a)*(a))
#define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
#define mem(a,b) memset(a,b,sizeof(a))
#define eps (1e-8)
#define J 10
#define mod 1000000007
#define MAX 0x7f7f7f7f
#define PI 3.14159265358979323
#define N 104
using namespace std;
typedef long long LL;
int cas,cass;
int n,m,lll,ans;
int c[N];
LL aans;
LL a[N][N];
LL f[N][N][N];
int main()
{
	#ifndef ONLINE_JUDGE
//	freopen("1.txt","r",stdin);
//	freopen("2.txt","w",stdout);
	#endif
	int i,j,k,l;
//	for(scanf("%d",&cass);cass;cass--)
//	for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
//	while(~scanf("%s",s+1))
	while(~scanf("%d",&n))
	{
		scanf("%d%d",&m,&cas);
		for(i=1;i<=n;i++)
			scanf("%d",&c[i]);
		for(i=1;i<=n;i++)
			for(j=1;j<=m;j++)
				scanf("%I64d",&a[i][j]);
		mem(f,0x7f);
		c[0]=m+1,f[0][0][m+1]=0;
		for(i=1;i<=n;i++)
		{
			if(c[i]!=0)
			{
				if(c[i-1]!=0)
				{
					for(j=1;j<=min(i,cas);j++)
					{
						if(c[i]!=c[i-1])
							f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j-1][c[i-1]]);
						else f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j][c[i-1]]);
					}
				}
				else//c[i]!=0 c[i-1]=0
				{
					for(j=1;j<=min(i,cas);j++)
					{
						for(k=1;k<=m;k++)
						{
							if(k!=c[i])
								f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j-1][k]);
							else f[i][j][c[i]]=min(f[i][j][c[i]],f[i-1][j][k]);
						}
					}
				}
			}
			else if(c[i-1]==0)//c[i]=0
			{
				for(j=1;j<=min(i,cas);j++)
				{
					for(k=1;k<=m;k++)
					{
						for(l=1;l<=m;l++)
						{
							if(l==k)f[i][j][k]=min(f[i][j][k],f[i-1][j][l]+a[i][k]);
							else f[i][j][k]=min(f[i][j][k],f[i-1][j-1][l]+a[i][k]);
						}
					}
				}
			}
			else//c[i]=0,c[i-1]!=0
			{
				for(j=1;j<=min(i,cas);j++)
				{
					for(k=1;k<=m;k++)
					{
						if(k!=c[i-1])
							f[i][j][k]=min(f[i][j][k],f[i-1][j-1][c[i-1]]+a[i][k]);
						else f[i][j][k]=min(f[i][j][k],f[i-1][j][c[i-1]]+a[i][k]);
					}
				}
			}
		}
		aans=f[0][0][0];
		for(i=1;i<=m;i++)
			aans=min(aans,f[n][cas][i]);
		if(aans==f[0][0][0])puts("-1");
		else printf("%I64d\n",aans);
	}
	return 0;
}
/*
//

//
*/


1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:60429次
    • 积分:2311
    • 等级:
    • 排名:第17789名
    • 原创:242篇
    • 转载:2篇
    • 译文:0篇
    • 评论:16条
    文章分类
    最新评论