SPFA 模板


复杂度:$O(nm)$

#include<bits/stdc++.h>

using namespace std;

#define ll long long
#define ull unsigned long long
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);

const int INF = 0x3f3f3f3f;
const int N=2002;

struct node {
    int to,val,next;
} e[N];

int n,m;
int head[N],dis[N],vis[N],cnt;

void add(int u,int v,int w) {
    e[++cnt].to=v;
    e[cnt].val=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}

void spfa(int s) {
    mem(dis,INF);
    mem(vis,0);
    dis[s]=0;
    vis[s]=1;
    queue<int> q;
    q.push(s);
    while(!q.empty()) {
        int u=q.front();
        vis[u]=0;
        q.pop();
        for(int i=head[u]; i; i=e[i].next) {
            int v=e[i].to;
            if(dis[v]>(dis[u]+e[i].val)) {
                dis[v]=dis[u]+e[i].val;
                if(!vis[v]) {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }

}
int main() {
    cin>>n>>m;
    memset(head,0,sizeof(head));
    cnt=1;
    for(int i=1; i<=m; i++) {
        int u,v,w;
        cin>>u>>v>>w;
        add(u,v,w);
        add(v,u,w);
    }
    spfa(1);
    for(int i=1;i<=n;i++)
        cout<<dis[i]<<" ";
}

文章作者: 小凡
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 小凡 !
评论
  目录
隐藏
{% if theme.sakura.enable %}{% endif %}