Invitation Cards POJ - 1511

最新推荐文章于 2022-02-17 13:15:00 发布

原创最新推荐文章于 2022-02-17 13:15:00 发布 · 235 阅读

CC 4.0 BY-SA版权

本文深入探讨了SPFA（Shortest Path Faster Algorithm）算法在图论中的应用，特别是求解从一个源点到所有其他点的最短路径问题。通过实例代码详细解析了算法的实现过程，包括节点初始化、队列操作以及距离更新等关键步骤，适用于理解并掌握SPFA算法原理及其在实际问题中的运用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//求1到每个点的最短路及每个点到一的最短路之和
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
int n,m;
const int inf=1<<30;
long long ans;
vector<pair<int,int> >e[1000006];
int a[1000006],b[1000006],c[1000006];
long long  dis[1000006];
bool vis[1000006];
void spfa()
{
    fill(dis,dis+n+1,inf);
    dis[1]=0;
    queue<int>q;
    q.push(1);
    vis[1]=1;
    while(!q.empty())
    {
        int s=q.front();
        q.pop();
        vis[s]=0;
        for(int i=0; i<e[s].size(); i++)
        {
            int v=e[s][i].first;
            int w=e[s][i].second;
            if(dis[v]>dis[s]+w)
            {
                dis[v]=dis[s]+w;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
    for(int i=1; i<=n; i++)
    {
        ans+=dis[i];
    }

}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        ans=0;
        scanf("%d%d",&n,&m);
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d%d",&a[i],&b[i],&c[i]);
            e[a[i]].push_back(make_pair(b[i],c[i]));
        }
        spfa();
        for(int i=1; i<=n; i++)
            e[i].clear();
        for(int i=1; i<=m; i++)
        {
            e[b[i]].push_back(make_pair(a[i],c[i]));
        }
        spfa();
        printf("%lld\n",ans);
        for(int i=1; i<=n; i++)
            e[i].clear();
    }
    return 0;
}