NYOJ203 三国志【01背包 + 最短路径】

三国志

时间限制：3000 ms  |  内存限制：65535 KB

难度：5

描述

《三国志》是一款很经典的经营策略类游戏。我们的小白同学是这款游戏的忠实玩家。现在他把游戏简化一下，地图上只有他一方势力，现在他只有一个城池，而他周边有一些无人占的空城，但是这些空城中有很多不同数量的同种财宝。我们的小白同学虎视眈眈的看着这些城池中的财宝。

按照游戏的规则，他只要指派一名武将攻占这座城池，里面的财宝就归他所有了。不过一量攻占这座城池，我们的武将就要留守，不能撤回。因为我们的小白手下有无数的武将，所以他不在乎这些。

从小白的城池派出的武将，每走一公理的距离就要消耗一石的粮食，而他手上的粮食是有限的。现在小白统计出了地图上城池间的道路，这些道路都是双向的，他想请你帮忙计算出他能得到 的最多的财宝数量。我们用城池的编号代表城池，规定小白所在的城池为0号城池，其他的城池从1号开始计数。

输入

本题包含多组数据：
首先，是一个整数T(1<=T<=20)，代表数据的组数
然后，下面是T组测试数据。对于每组数据包含三行：
第一行：三个数字S,N,M
（1<=S<=1000000,1<=N<=100,1<=M<=10000)
S代表他手中的粮食（石），N代表城池个数，M代表道路条数。
第二行：包含M个三元组行 Ai,Bi,Ci(1<=A,B<=N,1<=C<=100)。
代表Ai,Bi两城池间的道路长度为Ci(公里）。
第三行：包含N个元素，Vi代表第i个城池中的财宝数量。(1<=V<=100)

输出

每组输出各占一行，输出仅一个整数，表示小白能得到的最大财富值。

样例输入

2
10 1 1
0 1 3
2
5 2 3
0 1 2 0 2 4 1 2 1
2 3

样例输出

2
5

来源

郑州大学校赛题目

上传者

张云聪

问题连接：题目203 三国志（NYOJ）

解题思路：最短路径+0-1背包，先用Dijkstra算法求出源点0到其他结点的需要的花费，这样就成了0-1背包问题了。程序中dist[i]记录到第i个城市的代价，c[i]记录第i个城市的价值。

AC的C++程序：

#include<iostream>
#include<vector>
#include<queue> 
#include<cstdio>
#include<cstring>

using namespace std;
const int INF=0x3f3f3f3f;
const int N=105;

struct Edge{//边 
	int v,w;//起点到终点v的花费为w 
	Edge(int v,int w):v(v),w(w){}
};

struct Node{
	int u,w;//源点到编号为u的结点的最短路径为w
	Node(){}
	Node(int u,int w):u(u),w(w){}
	bool operator<(const Node &a)const//使用优先队列所以将小于重载为大于含义 
	{
		return w>a.w;
	} 
};
vector<Edge>g[N];//图的邻接表表示 
bool vis[N];
int dist[N];
int dp[1000005];//0-1背包 

void dijkstra(int s)//源点为s
{
	priority_queue<Node>q;
	memset(vis,false,sizeof(vis));
	memset(dist,INF,sizeof(dist));
	dist[s]=0;
	q.push(Node(s,0));
	while(!q.empty()){
		Node e=q.top();
		q.pop();
		int u=e.u;
		if(!vis[u]){
			vis[u]=true;
			int num=g[u].size();
			for(int i=0;i<num;i++){
				int v=g[u][i].v;
				if(!vis[v]){
					if(dist[u]+g[u][i].w<dist[v]){
						dist[v]=dist[u]+g[u][i].w;
						q.push(Node(v,dist[v]));
					}
				}
			}
		}
	}
}

int main()
{
	int T,s,n,m,a,b,c;
	scanf("%d",&T);
	while(T--){
		scanf("%d%d%d",&s,&n,&m);
		for(int i=0;i<=n;i++)
		  g[i].clear();
		while(m--){
			scanf("%d%d%d",&a,&b,&c);
			g[a].push_back(Edge(b,c));
			g[b].push_back(Edge(a,c));
		}
		dijkstra(0);
		//dist[i]记录到第i个城市的代价
		int c[N];//记录第i个城市的价值
		for(int i=1;i<=n;i++)
		  scanf("%d",&c[i]); 
		//0-1背包问题
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++)
		  for(int j=s;j>=dist[i];j--)
		    dp[j]=max(dp[j],dp[j-dist[i]]+c[i]);
		printf("%d\n",dp[s]); 
	}
	return 0;
}