hdu 5074 Hatsune Miku(dp)

dp[i][j]表示处理到第i位时最后一个数是j的最大价值。

转移就是dp[i][j]=max(dp[i-1][k]+cost[k][j]) (枚举k)


如果遇到给定的数,就把处理到这一位不是这个给定的数的情况全部处理为负无穷,因为是不合法的。初始化时如果第一个数就给定了也是这样处理。

代码:

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#define INF (-1000000)
#define LL long long
using namespace std;
int dp[105][55];
int cost[55][55];
int p[105];
int n,m;
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++){
			for(int j=1;j<=m;j++){
				scanf("%d",&cost[i][j]);
			}
		}
		for(int i=1;i<=n;i++)
			scanf("%d",&p[i]);
		memset(dp,0,sizeof(dp));
		if(p[1]!=-1){
			for(int i=1;i<=m;i++){
				dp[1][i]=INF;
			}
			dp[1][p[1]]=0;
		}
		else {
			for(int i=1;i<=m;i++){
			   dp[1][m]=0;
			}
		}
		
		for(int i=2;i<=n;i++){
			if(p[i]!=-1){
				for(int j=1;j<=m;j++){
					dp[i][p[i]]=max(dp[i-1][j]+cost[j][p[i]],dp[i][p[i]]);
				}
				for(int j=1;j<=m;j++){
					if(j!=p[i]) dp[i][j]=INF;
				}
			}
			else{
				for(int j=1;j<=m;j++){
					for(int k=1;k<=m;k++){
						dp[i][k]=max(dp[i-1][j]+cost[j][k],dp[i][k]);
					}
				}
			}
		}
		int res=0;
		for(int i=1;i<=m;i++){
			res=max(res,dp[n][i]);
		}
		cout<<res<<endl;
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值