悬关,求调

P4779 【模板】单源最短路径(标准版)

xinglili @ 2023-10-24 15:44:31

只能过最后两个点,不知道哪里有问题

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;

typedef pair<int,int> PII;
const int N=1e5+10,M=2e5+10;

int h[N],e[M],w[M],ne[M],idx;
void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

bool st[N];
int dis[N];

void dijkstra(int s)
{
    memset(dis,0x3f,sizeof(dis));
    dis[s]=0;

    priority_queue<PII,vector<PII>,greater<PII>> a;
    a.push({s,0});

    while(!a.empty())
    {
        PII t=a.top();
        a.pop();

        int num=t.first,distance=t.second;

        if(st[num])continue;
        st[num]=true;

        for(int i=h[num];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>distance+w[i])
            {
                dis[j]=distance+w[i];
                a.push({j,dis[j]});
            }
        }
    }
}

int main()
{
    memset(h,-1,sizeof(h));
    int n,m,s;
    cin>>n>>m>>s;

    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        cin>>a>>b>>c;
        add(a,b,c);
    }
    dijkstra(s);

    for(int i=1;i<=n;i++)cout<<dis[i]<<' ';

    return 0;
}

by only_a_speaker @ 2023-10-24 15:53:06

您好, std::pair 的排序第一关键字为 first ,第二关键字为 second 。建议调换两个成员的内容


by some_side @ 2023-10-24 16:50:49

写优先队列的时候别用pair啊,或者说要重载一下运算符再用。pair和pair之间比较大小事先比较 first,再比较 second,但是跑 dijkstra 的时候只用关心 second 的大小。所以呢,建议用结构体或者重载一下运算符。调换两个成员的内容的话,也行,但是总感觉怪怪的,一般都不这么写。


by xinglili @ 2023-10-24 19:16:25

没注意,刚学了这个算法~呜呜 都关注了


|