求助dij堆优化wa#1#3#4
查看原帖
求助dij堆优化wa#1#3#4
151647
sycqwq楼主2021/9/4 15:43

rt

#include<bits/stdc++.h>
using namespace std;
const int maxm=10e5+5;
int n,m,s,head[maxm<<1],tot;
struct node
{
    int v,nxt,w;
}e[maxm<<1];
int add(int u,int v,int w)
{
    e[++tot].v=v;
    e[tot].nxt=head[u];
    head[u]=tot;
    e[tot].w=w;
}
bool bk[maxm];
int dis[maxm];
// bool operator<(int x,int y)
// {
//     return dis[x]>dis[y]
// }
struct cmp
{
    bool operator()(int &s1,int &s2)
    {
        return dis[s1]>dis[s2];
    }
};
priority_queue<int,vector<int>,cmp> q;
void dij()
{
    dis[s]=0;
    q.push(1);
    while(!q.empty())
    {
        int x=q.top();
        q.pop();
        if(bk[x])
            continue;
        bk[x]=1;
        for(int i=head[x];i;i=e[i].nxt)
        {
            int v=e[i].v;
            if(!bk[v]&&dis[v]>dis[x]+e[i].w)
            {
                dis[v]=dis[x]+e[i].w;
                q.push(v);
            }
        }
    }
}
int main()
{
    cin>>n>>m>>s; 
    for(int i=1;i<=n;i++)
        dis[i]=INT_MAX;
    for(int i=1;i<=m;i++)
    {
        int x,y,w;
        cin>>x>>y>>w;
        add(x,y,w);
        add(y,x,w);
    }
    dij();
    for(int i=1;i<=n;i++)
        cout<<dis[i]<<' ';
    return 0;
}
2021/9/4 15:43
加载中...