PTA 挖地雷

12 篇文章 0 订阅

题目描述:

在一个地图上有n个地窖(n≤200),每个地窖中埋有一定数量的地雷。同时,给出地窖之间的连接路径,并规定路径都是单向的,且保证都是小序号地窖指向大序号地窖,也不存在可以从一个地窖出发经过若干地窖后又回到原来地窖的路径。某人可以从任意一处开始挖地雷,然后沿着指出的连接往下挖(仅能选择一条路径),当无连接时挖地雷工作结束。设计一个挖地雷的方案,使他能挖到最多的地雷。

输入格式:

第一行:地窖的个数;

第二行:为依次每个地窖地雷的个数;

下面若干行:

xi yi 表示从xi可到yi,xi<yi。

最后一行为"0 0"表示结束。

输出格式:

k1-k2−…−kv 挖地雷的顺序 挖到最多的雷。

输入样例:

6
5 10 20 5 4 5
1 2
1 4
2 4
3 4
4 5
4 6
5 6
0 0

输出样例:

3-4-5-6
34

思路:

本题是一个数字三角形模型的动态规划,定义 d p [ i ] dp[i] dp[i]为到第i个点能挖到的最多地雷数,因为题目保证都是小序号地窖指向大序号地窖,所以转移方程为 d p [ j ] = m a x ( d p [ j ] , d p [ i ] + p [ j ] ) dp[j]=max(dp[j],dp[i]+p[j]) dp[j]=max(dp[j],dp[i]+p[j])( p [ j ] 是 点 j 的 地 雷 数 p[j]是点j的地雷数 p[j]j).本题难点是记录路径.我们可以用一个数组path[i],每次dp方程转移时记录转移到该点的上一个点是谁( p a t h [ j ] = i path[j]=i path[j]=i),最后再找到能挖到最多地雷数的那个点,倒序输出path即可.

Hint:

  • 如果wa最后一个点可以试试这个测试样例:
 3
 1 2 100
 1 2
 0 0
  • 记得要将 d p [ i ] 初 始 化 为 p [ i ] dp[i]初始化为p[i] dp[i]p[i]

代码:

#include <iostream>
using namespace std;
int main(){
	int p[205],n,dp[205]={0},path[205]={0},pre[205];
	bool vis[205][205];
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>p[i];
		dp[i]=p[i];
	}
	int u,v;
	while(1){
		cin>>u>>v;
		if(u==0&&v==0)break;
		vis[u][v]=true;//vis[u][v]意思是u这个点可以到v这个点
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(vis[i][j]&&dp[i]+p[j]>dp[j]){
				dp[j]=dp[i]+p[j];
				path[j]=i;//记录路径
			}
		}
	}
	int k,ans=0;
	for(int i=1;i<=n;i++){
		if(dp[i]>ans){
			ans=dp[i];
			k=i;
		}
	}
	pre[0]=k;//pre[i]作用是倒序记录path
	int d=1;
	while(path[k]!=0){
		pre[d]=path[k];
		k=path[k];
		d++;
	}
	cout<<pre[d-1];
	for(int i=d-2;i>=0;i--){
		cout<<"-"<<pre[i];
	}
	cout<<'\n';
	cout<<ans;
} 
  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值