Dijkstra 模板


复杂度:$O((n+m)logm)$


#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ll long long
#define pii pair<int,int>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int INF = 0x3f3f3f3f;
const int N=2002;

struct edge {
    int next,to,w;
} e[N];

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

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

void dij(int s) {
    mem(dis,INF);
    mem(vis,0);
    dis[s]=0;
    priority_queue<pair<int,int> > q;
    q.push(make_pair(-dis[s],s));
    while(!q.empty()) {
        int u=q.top().second;
        q.pop();
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u]; i; i=e[i].next) {
            int v=e[i].to;
            if(dis[v]>dis[u]+e[i].w) {
                dis[v]=dis[u]+e[i].w;
                q.push(make_pair(-dis[v],v));
            }
        }
    }
}

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

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