最短路之SPFA模板

来源:互联网 发布:围棋判断胜负软件 编辑:程序博客网 时间:2024/05/03 13:25

一:邻接矩阵版本SPFA//如果要判断负环的话加一个记录入队的数组就行,当入队次数大于n的时候出现负环

int d[MAXN],vis[MAXN],w[MAXN][MAXN];int n;void SPFA(int s){    fill(d,d+n,INF);    d[s]=0;    queue<int> q;    q.push(s);    while(!q.empty())    {        int u=q.front();        q.pop();        for(int v=0; v<n; v++)        {            if(d[v]>d[u]+w[u][v])            {                d[v]=d[u]+w[u][v];                q.push(v);            }        }    }}

二:伪邻接表版SPFA

const int inf = 0x3f3f3f3f;int v[M],w[M],next[M],head[N],d[N],e;//head初始化为-1,模拟邻接表void addedge(int a,int b,int x)//主要如果是无向图的话,每条边要执行两次这个,a,b反过来就行{    v[e]=b;    w[e]=x;    next[e]=head[a];    head[a]=e++;}void SPFA(int s){    queue<int> q;    d[s]=0;    q.push(s);    while(!q.empty())    {        int u = q.front();        q.pop();        for(int i=head[u]; i!=-1; i=next[i])        {            if(d[v[i]]>d[u]+w[i])            {                d[v[i]]=d[u]+w[i];                q.push(v[i]);            }        }    }}

for(int i=0; i<m; i++)//输入一条带权值的无向边        {            cin>>a>>b>>x;            addedge(a,b,x);            addedge(b,a,x);        }

三:邻接表版

struct Edge{    int to,w,next;}E[maxn];void init(){    memset(head,-1,sizeof(head));    fill(d,d+n,INF);    size = 0;}void addedge(int u,int v,int x){    E[size].to = v;    E[size].w = x;    E[size].next = head[u];    head[u] = size++;}void SPFA(int s){    queue<int> q;    d[s]=0;    q.push(s);    while(!q.empty())    {        int u = q.front();        q.pop();        for(int i=head[u]; i!=-1; i=E[i].next)        {            if(d[E[i].to] > d[u] + E[i].w)            {                d[E[i].to] = d[u]+E[i].w;                q.push(E[i].to);            }        }    }}


0 0
原创粉丝点击