SDUT 3562 第七届山东省ACM省赛 Proxy （dijkstra最短路 + 反向边）

传送门：SDUT 3562

题目大意：

一个有编号 0~n+1 点的有向图。

1. 如果从 0 点到 n+1 点的最短路径不存在则输出 -1。

2. 如果最短路存在则输出最短路上离 0 点最近的点。

3. 如果存在多条最短路，则输出 节点编号小的离 0 点最近的点。

4. 如果该点恰好是 n+1 点，则输出 0.

思路：

一开始在想怎么记录最短路径上的点，我是先跑一遍最短路，然后用 dfs 求每条路径是否为最短路，如果是则记录离 0 最近的点，然鹅……MLE了……

正确的姿势是建反向边，即把边 a -> b，改为 b -> a，这样我们求一下从点 n+1 到点 0 的最短路，并用 dis[i] 记录 i 到 n+1 的最短距离。提前保存与 0 相邻的点，如果邻点 i 与 0 点的距离 + dis[i] = 最短路距离 的话，就说明点 i 是最短路径上的点，保存编号最小的即可。

代码：

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f

int n,m;
int dis[1010],vis[1010];
int mp[1010][1010];

int dijkstra(int v) 
{
	int i,j,min,pos;
	memset(vis,0,sizeof(vis)); //全标记为未处理	
	for(i=0;i<=n+1;i++)  //dis[i]表示当前节点与i的路径长度，初始化为最大 
		dis[i]=mp[v][i];
    dis[v]=0;
	vis[v]=1;
	//算法核心代码
	for(i=0;i<n+1;i++)
	{ //进行n+1次处理 
		min=inf;
		for(j=0;j<=n+1;j++) //寻找未处理的点中路径最短的 
			if(!vis[j] && dis[j]<min)
			{
				min=dis[j];
				pos=j;
			}
		vis[pos]=1;
		for(j=0;j<=n+1;j++) //用该节点更新其他未处理节点的最短路径 
			if(!vis[j]&&dis[j]>dis[pos]+mp[pos][j])
					dis[j]=dis[pos]+mp[pos][j];			
	}
}

int main()
{	
	int i,t,u,v,w,tol,ans,zero[200000];
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);	
		tol=0;	
		memset(mp,inf,sizeof(mp));
		for(i=0;i<=n+1;i++) mp[i][i]=0;
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			mp[v][u]=w; //反向建边 
			if(u==0) zero[tol++]=v; //保存与0相邻的点 
		}
		dijkstra(n+1);
		if(dis[0]==inf)
		{ //无最短路 
			printf("-1\n");
			continue;
		}
		ans=inf;
		for(i=0;i<tol;i++)
		{
			int v=zero[i]; 
			//如果 v 是最短路上的与0相邻的点 
			if(mp[v][0]+dis[v]==dis[0]&&v<ans) ans=v;
		}
		if(ans==n+1) printf("0\n");
		else printf("%d\n",ans);
	}
	return 0;
}