POJ 1791 Heavy Transportation（最短路变形 Dijkstra 以及堆优化）

题意：

有 $n$ 个点，$1$ 为起点，$n$ 为终点，问从 $1$ 走到 $n$ 所有可达通路中最小边的最大值，这道和昨天刚做的POJ 2253求所有可达通路中最大边的最小值差不多。 稍加修改dist[i]=max(dist[i],min(dist[tmp],mp[tmp][i]))

数据范围：$2<=n<=1000$

明显的Floyd $O(n^3)$ 复杂度太高了，先给出朴素的$O(n^2)$的Dijkstra。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<utility>
#include<algorithm>
using namespace std;
const int N = 1010;
const int inf = 0x3f3f3f3f;
int n,m;
int mp[N][N]; 
int dist[N]; //存 1~n 的距离
bool vis[N]; //是否确定 dist 了
int findMax()
{
	int mx=-inf,k=-1;
	for(int i=1;i<=n;i++)
	{
		if(!vis[i]&&dist[i]>mx)
		{
			mx=dist[i];
			k=i;
		}
	}
	return k;
}
void dij(int u)
{
	memset(vis,false,sizeof vis);
	for(int i=1;i<=n;i++)
		dist[i]=mp[u][i];
	dist[0]=0;
	vis[u]=true;
	
	for(int k=1;k<n;k++)
	{
		int tmp=findMax();  //找最大值的点
		vis[tmp]=true; //已被加入 T 集合
		for(int i=1;i<=n;i++)
			if(!vis[i])
				dist[i]=max(dist[i],min(dist[tmp],mp[tmp][i])); //可达通路中最小边的最大值
	}
}
int main()
{	
	int t,u,v,w;
	cin>>t;
	for(int k=1;k<=t;k++)
	{	
		scanf("%d %d",&n,&m);
		memset(mp,-inf,sizeof mp);
		for(int i=1;i<=m;i++)
		{
			scanf("%d %d %d",&u,&v,&w);
			mp[u][v]=mp[v][u]=w; //无向图
		}	
		dij(1);
		printf("Scenario #%d:\n%d\n\n",k,dist[n]);
	}	
	return 0;
}

邻接表 $O(nlogn)$

开始一直用邻接表做 RE 了十几次，硬是没找出错误，终于！！想到了有可能这题的测试数据 边给的很多，所以 链式前向星的边集范围要开到 $10^6$，哎，还不如直接用二维数组存储呢，少了好多麻烦（详细的坑见下面注释）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<utility>
#include<algorithm>
using namespace std;
const int N = 1e6+10;
const int inf = 0x3f3f3f3f;
typedef pair<int,int> pii; // first 存距离，second 存点
int n,m;
struct edge{
	int to,w,next;
}e[2*N];
int h[N],tot;
void add(int u,int v,int w)
{
	e[tot].to=v;
	e[tot].w=w;
	e[tot].next=h[u];
	h[u] = tot++;
} 

int dist[N];
bool vis[N];
priority_queue<pii> q;
void dij(int u)
{
	while(!q.empty())
		q.pop();		
	memset(dist,-inf,sizeof dist); // 开始 dist 初始化为 负无穷，因为最后要取最大值
	memset(vis,false,sizeof vis);
	for(int i=h[u];i!=-1;i=e[i].next) // 就是这里，考虑到初始化为负无穷，第一个点和所有点的距离要单独拿出来解决，否则第一轮循环的时候，以第一个点为中间点去更新它的邻接点的 dist 都会为 0，而实际上第一轮应该是 1 到其他点的距离值。
	{
		int j=e[i].to; 
		dist[j]=e[i].w;
		q.push(make_pair(dist[j],j));
	}
	dist[u]=0;
	vis[u]=true;
	
	while(!q.empty())
	{
		pii tmp=q.top(); // 优先队列默认 less 从大到小排列
		q.pop();
		int ver=tmp.second,distance=tmp.first;
		
		if(vis[ver]) continue; // dist 已被确定
		vis[ver]=true;

		for(int i=h[ver];i!=-1;i=e[i].next)
		{
			int j=e[i].to;
			if(dist[j] < min(distance,e[i].w)) 
			{
				dist[j] = min(distance,e[i].w);
				q.push(make_pair(dist[j],j));
			}		 
		}
	}
}
int main()
{	
	int t,u,v,w;
	cin>>t;
	for(int k=1;k<=t;k++)
	{	
		scanf("%d %d",&n,&m);
		memset(h,-1,sizeof h); // 邻接表的头结点重新初始化
		tot=0;   //边的总数也要重新初始化
		for(int i=1;i<=m;i++)
		{
			scanf("%d %d %d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}	
		dij(1);
		printf("Scenario #%d:\n%d\n\n",k,dist[n]);
	}	
	return 0;
} 

不过也就快了 $40ms$，但是内存大到爆啊，所以选用邻接表还是要慎重！