POJ1797--Heavy Transportation

本题可以用dijkstra做，但是dis数组表示的不是从起点到i点的最短距离，而是最大载量，然后再改变下松弛条件，就可以轻松AC。

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <math.h>
#define INF 1>>30
#define M 1005
#define Min(a,b) a<b?a:b
#define Max(a,b) a>b?a:b
using namespace std;

int n,m;
int dis[M];
int map[M][M];
bool visit[M];

void init()
{
	int i,j;
	for(i=0;i<M;i++)
	{
		for(j=0;j<M;j++)
		{
			map[i][j]=0;
		}
	}
}

void dijkstra()
{
	int i,j,max,flag;
	memset(visit,false,sizeof(visit));
	for(i=1;i<=n;i++)
		dis[i]=map[1][i];
	dis[1]=0;
	for(i=1;i<n;i++)
	{
		max=-1;
		for(j=1;j<=n;j++)
		{
			if(!visit[j] && dis[j]>max)
			{
				max=dis[j];
				flag=j;
			}
		}
		if(max==-1)
			break;
		visit[flag]=true;
		for(j=1;j<=n;j++)
		{
			dis[j]=Max(Min(dis[flag],map[flag][j]),dis[j]);
		}
	}
}

int main()
{
	int t,count,i;
	scanf("%d",&t);
	for(count=1;count<=t;count++)
	{
		scanf("%d%d",&n,&m);
		init();
		for(i=1;i<=m;i++)
		{
			int a,b,d;
			scanf("%d%d%d",&a,&b,&d);
			map[a][b]=map[b][a]=d;
		}
		printf("Scenario #%d:\n",count);
		dijkstra();
		printf("%d\n\n",dis[n]);
	}
}