道路与航线(单源最短路+拓扑排序中最短路)

题目

如果是一个有向无环图，那么这个图的最短路一定是拓扑序出的最短路，如果一个点入度不为0，那么最短路能去更新他的点一定是与他相连指向它的点，在这个点出队前，那么它的入度点一定都更新过它，所以它一定是最小值。

#include <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define x first
#define y second 
typedef pair<int,int> PII;
const int N=1e5+10;
int n,m,ecnt,r,p,s,item;
int head[25010];
int id[25010],ru[25010],chu[25010];
vector<int> block[25010];
int d[25010],vis[25010];
queue<int> q;
struct edge {
	int u,v,w,next;
} E[N<<1];


void add(int u,int v,int w) {
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].w=w;
	E[ecnt].next=head[u];
	head[u]=ecnt;
}


void dfs(int u,int it) {//dfs找出所有的团
	id[u]=it;//标记u为it团
	block[it].push_back(u);//加入团
	for(int i=head[u]; i; i=E[i].next) {//继续找出所有相邻的点，并且加入团
		int v=E[i].v;
		if(!id[v]) {
			dfs(v,it);
		}
	}
}

void dijkstra(int u) {
	priority_queue<PII,vector<PII>,greater<PII>>heap; 
	for(int i=0; i<block[u].size(); i++) {//放入堆中 
		heap.push({d[block[u][i]],block[u][i]});
	}
	
	while(!heap.empty()){
		auto u=heap.top();
		heap.pop();
	    if(vis[u.y])continue;
		vis[u.y]=1;
		for(int i=head[u.y];i;i=E[i].next){
			int v=E[i].v;
			if(d[v]>u.x+E[i].w){
				d[v]=u.x+E[i].w;
				if(id[u.y]==id[v]){//在一个团中 
					heap.push({d[v],v}); 
				} 
			}
			if(id[u.y]!=id[v]&&--ru[id[v]]==0){//不在一个团中 
				q.push(id[v]);//加入到拓扑序列中 
			} 
		} 
	} 
}


void topSort() {//按照拓扑排序求有向无环图最短路
	memset(d,INF,sizeof d);
	d[s]=0;//出发点所在的团之前的团都无法到达 

	for(int i=1; i<=item; i++) {
		if(!ru[i]) q.push(i);
	}
	while(!q.empty()) {
		int t=q.front();
		q.pop();
		dijkstra(t);
	}
}


int main() {


	cin>>n>>r>>p>>s;
	//读入道路
	for(int i=1; i<=r; i++) {
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
	//处理团
	for(int i=1; i<=n; i++) {
		if(!id[i]) {
			dfs(i,++item);
		}
	}
	//读入航向
	for(int i=1; i<=p; i++) {
		int u,v,w;
		cin>>u>>v>>w;
		ru[id[v]]++;//对应的团入度++
		chu[id[u]]++; //对应的团出度++
		add(u,v,w);
	}
    
    topSort();
	for(int i=1;i<=n;i++){
		if(d[i]>=INF/2)puts("NO PATH");//因为有负数所以大于无穷大的一般就无法到达 
		else printf("%d\n",d[i]); 
	} 














	return 0;
}