最短路的应用

//P3371 【模板】单源最短路径（弱化版）
#include<bits/stdc++.h>

using namespace std;
const int N=1000010;
int dist[N],h[N],e[N],ne[N],w[N],idx;
bool st[N];
int n,m,s;

void add(int a,int b,int c)
{
	e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

void dijkstra()
{
	memset(dist,0x3f,sizeof dist);
	dist[s]=0;
	
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> heap;
	heap.push({0,s});
	/*
	模板
	dist[1]=0;
	heap.push({0,1});  

	改良
	dist[s]=0;
	heap.push({0,s});
	这两步是实现求 从s到其他点的距离  dist[1] 为从s到1的距离
	*/
	
	while(heap.size())
	{
		auto t=heap.top();
		heap.pop();
		
		int ver=t.second,distance=t.first;
		if(st[ver]) continue;
		st[ver]=true;
		
		for(int i=h[ver];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>w[i]+distance)
			{
				dist[j]=w[i]+distance;
				heap.push({dist[j],j});
			}
		}
	}
}

int main()
{
	memset(h,-1,sizeof h);
	cin>>n>>m>>s;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
	}
	dijkstra();
	long long aa=pow(2,31)-1;
	
	for(int i=1;i<=n;i++)
	{
		if(dist[i]==0x3f3f3f3f)  cout<<aa<<" ";
		else cout<<dist[i]<<" ";
	}
	return 0;
}

//P1629 邮递员送信
/*
在求返回路径的距离的时候
我们需要求出 对于 2~n 这些点 他们到 1的距离

因此，我们把a与b的方向改变
我们就只需要一次dijkstra,d[s] 就变成了 从 s 到 1的最短距离 
*/
#include<bits/stdc++.h>

using namespace std;
const int N=1000010;
int dist[N],h[N],e[N],ne[N],w[N],idx;
bool st[N];
int n,m;
int a,b,c;

struct Edge
{
	int a;
	int b;
	int c;
}edges[N];

void add(int a,int b,int c)
{
	e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

void dijkstra(int u)
{
	memset(dist,0x3f,sizeof dist);
	dist[u]=0;
	
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> heap;
	heap.push({0,u});
	
	while(heap.size())
	{
		auto t=heap.top();
		heap.pop();
		
		int ver=t.second,distance=t.first;
		if(st[ver]) continue;
		st[ver]=true;
		
		for(int i=h[ver];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>w[i]+distance)
			{
				dist[j]=w[i]+distance;
				heap.push({dist[j],j});
			}
		}
	}
}

int main()
{
	long long ans=0;
	memset(h,-1,sizeof h);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		edges[i]={b,a,c};
		add(a,b,c);
	}
	dijkstra(1);
	for(int i=2;i<=n;i++) ans+=dist[i];
	
	memset(h,-1,sizeof h);
	memset(st,false,sizeof st);
	
	for(int i=1;i<=m;i++)
	{
		add(edges[i].a,edges[i].b,edges[i].c);// 改变点的方向
	}
	
	dijkstra(1);
	for(int i=2;i<=n;i++) ans+=dist[i];
	
	cout<<ans; 
	return 0;
}