单源最短路SPFA（模板）

/*
单源最短路SPFA
时间复杂度O(kE)
这个是队列实现，有时候改成栈实现会更加快，很容易修改
这个复杂度是不定的
*/

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<fstream>
#include<string.h>
using namespace std;

const int MAXN=1010;
const int INF=0x3f3f3f3f;

struct Edge
{
int v;
int cost;
Edge(int _v=0,int _cost=0): v(_v),cost(_cost){}
};

vector<Edge> E[MAXN];

void addedge(int u,int v,int w)
{
E[u].push_back(Edge(v,w));
}

bool vis[MAXN];
int cnt[MAXN];
int dist[MAXN];

bool SPFA(int start,int n)
{
memset(vis,false,sizeof(vis));
for(int i=1;i<=n;i++)
dist[i]=INF;
vis[start]=true;
dist[start]=0;
queue<int> que;
while(!que.empty()) que.pop();
que.push(start);
memset(cnt,0,sizeof(cnt));
cnt[start]=1;
while(!que.empty())
{
int u=que.front();
que.pop();
vis[u]=false;
for(int i=0;i<E[u].size();i++)
{
int v=E[u][i].v;
if(dist[v]>dist[u]+E[u][i].cost)
{
dist[v]=dist[u]+E[u][i].cost;
if(!vis[v])
{
vis[v]=true;
que.push(v);
if(++cnt[v]>n)
return false;
//cnt[i]为入队列次数，用来判断是否存在负环回路
}
}
}
}
return true;
}

int main()
{
freopen("in.txt","r",stdin);
int n;
int e;
cin>>n>>e;
int u,v,w;
for(int i=1;i<=e;i++)
{
cin>>u>>v>>w;
addedge(u,v,w);
}
bool flag=SPFA(1,n);
if(!flag)
cout<<"have negative circle"<<endl;
else
cout<<"no negative circle"<<endl;
for(int i=1;i<=n;i++)
cout<<dist[i]<<" ";
cout<<endl;
return 0;
}