最短路spfa、dijkstra模板代码

#include<bits/stdc++.h>
using namespace std;
#define int long long 
#define PII pair<int,int> 
struct node{
	int u,v,w;
};
const int N=100010,M=100010,inf=(1<<31)-1;
vector<node> e[M];
int dis[N],f[N],cnt[N],cnt1[N];
int n,m;
void add(int u,int v,int w){
	e[u].push_back({u,v,w});
}
void dij(int x,int (&dis)[N],int (&cnt)[N],int (&f)[N]){
	memset(f,0,sizeof(f));
	for(int i=1;i<=n;i++){
		dis[i]=inf;
	}
	priority_queue<PII,vector<PII>,greater<PII>> q;
	q.push({0,x});
	dis[x]=0;
	while(q.size()){
		PII t=q.top();
		q.pop();
		if(f[t.second]){
			continue;
		}
		f[t.second]=1;
		for(auto i:e[t.second]){
			if(dis[i.u]+i.w==dis[i.v]){
				cnt[i.v]+=cnt[i.u];
			}
			if(dis[i.u]+i.w<dis[i.v]){
				dis[i.v]=dis[i.u]+i.w;
				q.push({dis[i.v],i.v});
			}
		}
	}
}
bool spfa(int x,int (&dis)[N],int (&cnt1)[N],int (&cnt2)[N],int (&f)[N]){
	queue<int> q;
	memset(f,0,sizeof(f));
	for(int i=1;i<=n;i++){
		dis[i]=inf;
	}
	q.push(x);
	f[x]=1;
	dis[x]=0;
	while(q.size()){
		int t=q.front();
		q.pop();
		f[t]=0;
		for(auto i:e[t]){
			if(dis[i.v]==dis[i.u]+i.w){
				cnt2[i.v]+=cnt2[i.u];
			}
			if(dis[i.v]>dis[i.u]+i.w){
				dis[i.v]=dis[i.u]+i.w;
				if(f[i.v]==0){
					q.push(i.v);
					f[i.v]=1;
				}
				cnt1[i.v]=cnt1[t]+1;
				if(cnt1[i.v]>=n){
					return false;
				}
			}
		}
	} 
	return true;
}
signed main(){
	int s;
	scanf("%lld%lld%lld",&n,&m,&s);
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%lld%lld%lld",&u,&v,&w);
		add(u,v,w);
	}
	
	return 0;
}