为啥spfa会超时,最坏不就是o(nm)吗
查看原帖
为啥spfa会超时,最坏不就是o(nm)吗
644011
RainandBOW楼主2022/7/27 09:57
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int a[N],b[N],c[N];
int h[N],e[N],ne[N],idx,w[N];
int dist[N];
int n,m;
bool st[N];
void add(int x,int y,int z)
{
    w[idx]=z,e[idx]=y,ne[idx]=h[x],h[x]=idx++;
}
void spfa()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    queue<int>q;
    q.push(1);
    st[1]=true;
    while(q.size())
    {
        int t=q.front();
        q.pop();
        st[t]=false;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[t]+w[i])
            {
                dist[j]=dist[t]+w[i];
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m;
    memset(h, -1, sizeof h);
    for(int i=1;i<=m;i++)
    {
        cin>>a[i]>>b[i]>>c[i];
        add(a[i],b[i],c[i]);
    }
    spfa();
    int ans=0;
    for(int i=2;i<=n;i++) ans+=dist[i];
    memset(h, -1, sizeof h);
    for(int i=1;i<=m;i++)
    {
        add(b[i],a[i],c[i]);
    }
    spfa();
    for(int i=2;i<=n;i++) ans+=dist[i];
    cout<<ans<<endl;
    return 0;
}
2022/7/27 09:57
加载中...