SPFA模板

JUPAFA SPFA(洛谷P1339 USACO09OC热浪Heat Wave)

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std; 
const int INF=1<<30,maxn=50004,maxm=100006;
struct Edge{
	int to,w,next;
}edge[maxm];
int cnt,n,m,front[maxn],s,t;
inline void Add(int u,int v,int w){
	++cnt;edge[cnt].to=v;edge[cnt].w=w;edge[cnt].next=front[u];front[u]=cnt;
}
void SPFA(){
	int vis[maxn],d[maxn];
	queue<int>q;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++) d[i]=INF;
	d[s]=0;vis[s]=1;q.push(s);
	while(!q.empty()){
		int u=q.front();q.pop();vis[u]=0;
		for(int i=front[u];i>=0;i=edge[i].next){
			int v=edge[i].to,w=edge[i].w;
			if(d[v]>d[u]+w){
				d[v]=d[u]+w;
				if(!vis[v]){
					q.push(v);vis[v]=1;
				}
			}
		}
	}
	printf("%d\n",d[t]);
}
int main(){
	memset(front,-1,sizeof(front));
	scanf("%d%d%d%d",&n,&m,&s,&t);
	int u,v,w;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&u,&v,&w);
		Add(u,v,w);
		Add(v,u,w);
	}
	SPFA();
	return 0;
}