NYOJ-203三国志【最短路+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

这道题用了 Dijkstra算法求最短路径。公里数就是每条边的权值，然后将求得的dist[x]的值（也就是从起点到那座城最小的耗米量）。因为米量有限所以只能选择能最大聚宝的一些点，dist[x]的值再与每座城的宝藏价值v[y]当做01背包问题解决，求出的结果就是最大聚宝量。这个题和两种好难得算法嵌套使用，一个不对就AC不了，擦~~~

#include<stdio.h>
#include<stdio.h>
int load[109][109];
int v[109];
bool used[109];
int food[109][109];
int dist[109];
int ans[1000099];
int inf=99999;
int minx(int a,int b)
{
	if(a<b)
		return a;
	return b;
}
void Dijkstra(int n)
{
	int i,j;
	int q;
	int min;
	int loc=0;
	used[0]=1;
	dist[0]=0;
	i=n+1;
	for(j=1;j<=n;j++)//初始化
	{dist[j]=inf+1;used[j]=0;}
	while(i--)
	{
		min=inf;
		for(j=0;j<=n;j++)
		{//找新加入点
			if(!used[j]&&min>dist[j])
			{
				min=dist[j];loc=j;
			}
		}
		used[loc]=1;
		for(j=1;j<=load[loc][0];j++)
		{//更新路径
			q=load[loc][j];
			if(!used[q])
			{dist[q]=minx(dist[q],dist[loc]+food[loc][q]);}
		}
	}
//	for(i=1;i<=n;i++)
//		printf("%d ",dist[i]);
}
void _01backpack(int n,int s)
{
	int i,j;
	for(i=0;i<=s;i++)
		ans[i]=0;
	for(i=1;i<=n;i++)
	{
		for(j=s;j>=dist[i];j--)//注意只能是从S->0 不能是0->s
		{
				
			if(ans[j]<ans[j-dist[i]]+v[i])
				ans[j]=ans[j-dist[i]]+v[i];
		}
	}
	printf("%d\n",ans[s]);
}
int main()
{
	int T;
	int s,n,m;
	int i,j;
	int a,b,c;
	int q;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&s,&n,&m);
		for(i=0;i<=n;i++)
		{
			load[i][0]=0;
			for(j=0;j<=n;j++)
				food[i][j]=0;
		}
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(food[a][b]==0||food[a][b]>c)//因为可能有重边，所以取权最小的边
			{//建立邻接表
			q=++load[a][0];load[a][q]=b;
			q=++load[b][0];load[b][q]=a;
			food[a][b]=food[b][a]=c;
			}
		}
		for(i=1;i<=n;i++)
		{scanf("%d",&j);v[i]=j;}
		Dijkstra(n);
		 _01backpack(n,s);
	}
	return 0;
}