codeforces Round #369 (Div. 2) C. Coloring Trees (三维DP)


C. Coloring Trees
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

ZS the Coder and Chris the Baboon has arrived at Udayland! They walked in the park where n trees grow. They decided to be naughty and color the trees in the park. The trees are numbered with integers from 1 to n from left to right.

Initially, tree i has color ci. ZS the Coder and Chris the Baboon recognizes only m different colors, so 0 ≤ ci ≤ m, where ci = 0 means that tree i is uncolored.

ZS the Coder and Chris the Baboon decides to color only the uncolored trees, i.e. the trees with ci = 0. They can color each of them them in any of the m colors from 1 to m. Coloring the i-th tree with color j requires exactly pi, j litres of paint.

The two friends define the beauty of a coloring of the trees as the minimum number of contiguous groups (each group contains some subsegment of trees) you can split all the n trees into so that each group contains trees of the same color. For example, if the colors of the trees from left to right are 2, 1, 1, 1, 3, 2, 2, 3, 1, 3, the beauty of the coloring is 7, since we can partition the trees into 7contiguous groups of the same color : {2}, {1, 1, 1}, {3}, {2, 2}, {3}, {1}, {3}.

ZS the Coder and Chris the Baboon wants to color all uncolored trees so that the beauty of the coloring is exactly k. They need your help to determine the minimum amount of paint (in litres) needed to finish the job.

Please note that the friends can't color the trees that are already colored.

Input

The first line contains three integers, nm and k (1 ≤ k ≤ n ≤ 1001 ≤ m ≤ 100) — the number of trees, number of colors and beauty of the resulting coloring respectively.

The second line contains n integers c1, c2, ..., cn (0 ≤ ci ≤ m), the initial colors of the trees. ci equals to 0 if the tree number i is uncolored, otherwise the i-th tree has color ci.

Then n lines follow. Each of them contains m integers. The j-th number on the i-th of them line denotes pi, j (1 ≤ pi, j ≤ 109) — the amount of litres the friends need to color i-th tree with color jpi, j's are specified even for the initially colored trees, but such trees still can't be colored.

Output

Print a single integer, the minimum amount of paint needed to color the trees. If there are no valid tree colorings of beauty k, print  - 1.

Examples
input
3 2 2
0 0 0
1 2
3 4
5 6
output
10
input
3 2 2
2 1 2
1 3
2 4
3 5
output
-1
input
3 2 2
2 0 0
1 3
2 4
3 5
output
5
input
3 2 3
2 1 2
1 3
2 4
3 5
output
0
Note

In the first sample case, coloring the trees with colors 2, 1, 1 minimizes the amount of paint used, which equals to 2 + 3 + 5 = 10. Note that 1, 1, 1 would not be valid because the beauty of such coloring equals to 1 ({1, 1, 1} is a way to group the trees into a single group of the same color).

In the second sample case, all the trees are colored, but the beauty of the coloring is 3, so there is no valid coloring, and the answer is - 1.

In the last sample case, all the trees are colored and the beauty of the coloring matches k, so no paint is used and the answer is 0.

题意:有n棵树,让你分成k段,数字0代表该树没没有进行染色,需要你从1~m

种颜色中选择一个为其染色,染色的过程中需要消耗代价,求最小代价(已有颜色的树

不必再染色)

题解:典型DP

dp[i][j][k]表示当前轮到第i棵树选择第j种颜色且分成k段的最小代价

状态转移详见代码:

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<limits.h>
#include<algorithm>
#include<stack>
#include<vector>
#include<queue>
#include<math.h>
using namespace std;
#define maxn 105
#define inf 1e18  
int n,m,k;
__int64 a[maxn];
__int64 dp[maxn][maxn][maxn];
__int64 cost[maxn][maxn];
int  main()
{
	  int i,j,t,x;
	  scanf("%d%d%d",&n,&m,&k);
		for(i=1;i<=n;i++)
		  scanf("%I64d",&a[i]);
		for(i=1;i<=n;i++) 
		   for(j=1;j<=m;j++)
		      scanf("%I64d",&cost[i][j]);
		for(i=1;i<=n;i++)
		  for(j=1;j<=m;j++)
		    for(t=0;t<=n;t++)
		      dp[i][j][t]=inf;
		for(i=1;i<=n;i++)
		  for(t=1;t<=i;t++)
		  {
		  	if(a[i])
		  	{
		  	     dp[i][a[i]][t]=dp[i-1][a[i]][t];
				 for(j=1;j<=m;j++)
				  if(a[i]!=j)
				    dp[i][a[i]][t]=min(dp[i][a[i]][t],dp[i-1][j][t-1]);	
			}
			else
			{
				for(j=1;j<=m;j++)
				{
				  for(x=1;x<=m;x++)
				  {
				  	   if(j==x)
				  	     dp[i][x][t]=min(dp[i][x][t],dp[i-1][j][t]+cost[i][x]);
				  	   else
				  	     dp[i][x][t]=min(dp[i][x][t],dp[i-1][j][t-1]+cost[i][x]);
				  }
			   }
		    }
		  }
		__int64 ans=inf;
		for(i=1;i<=m;i++)
		   ans=min(ans,dp[n][i][k]);
		if(ans>=inf)
		   printf("-1\n");
		else
		   printf("%I64d\n",ans);
}


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值