dijkstra 建立虚拟源点求最短路

题目:
有 N 个村庄,编号 1 到 N。

村庄之间有 M 条无向道路,第 i 条道路连接村庄 ai 和村庄 bi,长度是 ci。

所有村庄都是连通的。

共有 K 个村庄有商店,第 j 个有商店的村庄编号是 xj。

然后给出 Q 个询问,第 k 个询问给出一个村庄的编号 yk,问该村庄距离最近的商店有多远?

输入格式
第一行包含两个整数 N,M。

接下来 M 行,每行包含三个整数 ai,bi,ci,表示第 i 条道路连接村庄 ai 和村庄 bi,长度是 ci。

再一行包含整数 K。

接下来 K 行,每行包含一个整数 xj,表示第 j 个有商店的村庄编号是 xj。

再一行包含整数 Q。

接下来 Q 行,每行包含一个整数 yk,表示询问编号为 yk 的村庄与其距离最近的商店之间的距离。

输出格式
对于每个询问,输出该询问的结果。

数据范围
2≤N≤105,
N−1≤M≤min(N(N−1)2,105),
1≤Q≤105,
1≤K≤N,
1≤ci≤10000

建立虚拟源点的步骤可以省去,直接将所有的起点入队,距离设置为0。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1e5+10,M = 1e5+10;
typedef pair<int,int> PII;
int e[2*M],ne[2*M],h[N],w[2*M],idx;
int dist[N];
int id[N];
bool st[N];
int n,m;

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

int main()
{
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    for(int i = 0; i<m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    int k;
    scanf("%d",&k);
    memset(dist,0x3f,sizeof dist);
    priority_queue<PII,vector<PII>,greater<PII> >heap;
    for(int i = 0;i<k;i++)
    {
        int x;
        scanf("%d",&x);
        dist[x] = 0;
        heap.push({0,x});
    }
    
    while(heap.size())
    {
        auto t= heap.top();
        heap.pop();
        int ver = t.second,distance = t.first;
        if(st[ver]) continue;
        st[ver] = true;
        for(int i = h[ver];~i;i=ne[i])
        {
            int j =e[i];
            if(dist[j]>distance+w[i])
            {
                dist[j] = distance+w[i];
                heap.push({dist[j],j});
            }
        }
    }
    int q;
    scanf("%d",&q);
    while(q--)
    {
        int x;
        scanf("%d",&x);
        printf("%d\n",dist[x]);
    }
    return 0;
}
posted @ 2023-03-03 11:04  希望上课能听懂  阅读(48)  评论(0)    收藏  举报