畅通工程续 hdu 杭电1874 dijkstra堆优化版最短路问题

Posted 2021-06-10 要坚持写博客

原题链接

堆优化版dijkstra 求最短路问题，这题目最需要注意的就是初始化，因为是多组案例。类似模板题。

A的第一道最短路问题，真爽！！！

#include<iostream>
#include<cstring>
#include<queue>
#include<set>
#include<algorithm>
using namespace std;
typedef pair<int,int> pll;
const int N=210,M=1010;
int h[M],e[M],ne[M],w[M],idx;
int n,m;
int dist[M];
bool st[M];

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

int dijkstra(int s,int x)
{
    memset(st,false,sizeof st);

    memset(dist, 0x3f, sizeof dist);
    dist[s]=0;
    priority_queue<pll,vector<pll>,greater<pll>>heap;
    heap.push({0,s});

    while(heap.size())
    {
        pll t=heap.top();
        heap.pop();

        int ver=t.second;

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

        for(int i=h[ver];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[ver]+w[i])
            {
                dist[j] = dist[ver] + w[i];
                heap.push({dist[j], j});
            }
        }
    }
    return dist[x];
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        idx=0;//这东西刚开始没发现，W了两次终于发现了
        cin>>m;
        memset(h,-1,sizeof h);
        for(int i=0;i<m;i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            add(a,b,c);
            add(b,a,c);
        }
        int s,x;
        cin>>s>>x;
        int r=dijkstra(s,x);

        if(r>=0x3f3f3f3f/2) cout<<"-1"<<endl;
        else cout<<r<<endl;
    }
    return 0;
}