SPFA(模板,邻接矩阵)

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
#define MAX 0x3f3f3f3f
int dis[2200],vis[2200];
queue<int> q;
int map[2000][2000];
void SPFA(int s,int n)
{
    dis[s]=0;
    vis[s]=1;
    q.push(s);


    while(!q.empty())
    {


        int t=q.front();
        q.pop();
        vis[t]=0;
        for(int i=0;i<n;i++)
        {
            int y=map[t][i];
            if(dis[t]+y<dis[i])
            {
                dis[i]=dis[t]+y;
                if(vis[i]==0)
                {
                    q.push(i);
                    vis[i]=1;
                }
            }
        }
    }
}
int main()
{
    for(;;)
    {
        memset(dis,MAX,sizeof(dis));
        memset(vis,0,sizeof(vis));
        memset(map,MAX,sizeof(map));
        int i,j,k,a,b,c,n,m,ans,start,endd;
        if(scanf("%d %d",&n,&m)==EOF)
            break;


        for(i=1; i<=m; i++)
        {
            scanf("%d %d %d",&a,&b,&c);
            if (c<map[a][b])
                map[a][b]=map[b][a]=c;


        }
        scanf("%d %d",&start,&endd);
        SPFA(start,n);
        ans=dis[endd];
        if(ans<MAX)
            printf("%d\n",ans);
        else
            printf("-1\n");
    }








    return 0;
}

猜你喜欢

转载自blog.csdn.net/septembre_/article/details/80670955