SPFA和前向星(适合边和点多的情况)

<code>这里写图片描述</code>

```
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
struct node
{
    int v,w;
    int next;
}edge[4000010];
int dis[500010];
int vis[500010];
int head[500010];
int cnt;
int n;

void add(int u,int v,int w) // 这里是倒着储存的(并且是起点相同的,第一个next指向-1)
{
    edge[cnt].v=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void SPFA(int s,int e)
{
     int i;
    queue<int>q;
    memset(vis,0,sizeof(vis));
    for(i=1; i<=n; i++)
    {
        dis[i]=inf;
    }
    dis[s]=0;
    vis[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            if(dis[v]>dis[u]+edge[i].w)
            {
                dis[v]=dis[u]+edge[i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);//这里可以用来判负环
                }
            }
        }
    }
}
int main()
{
    int m;
    int u,v,w;
    int s,e;
    while(~scanf("%d %d",&n,&m))
    {
        memset(head,-1,sizeof(head));
        cnt=1;

        while(m--)
        {

            scanf("%d %d %d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        scanf("%d %d",&s,&e);
        SPFA(s,e);
        printf("%d\n",dis[e]);
    }
    return 0;
}


“`

猜你喜欢

转载自blog.csdn.net/qq_42854569/article/details/81806277