《P11966 [GESP202503 八级] 上学》

题目背景

对应的选择、判断题：试题 - GESP 202503 C++ 八级 - 洛谷有题

题目描述

C 城可以视为由 n 个结点与 m 条边组成的无向图。 这些结点依次以 1,2,...,n 标号，边依次以 1≤i≤m 连接边号为 ui​ 与 vi​ 的结点，长度为 li​ 米。

小 A 的学校坐落在 C 城的编号为 s 的结点。小 A 的同学们共有 q 位，他们想在保证不迟到的前提下，每天尽可能晚地出门上学。但同学们并不会计算从家需要多久才能到学校，于是找到了聪明的小 A。第 i 位同学 (1≤i≤q) 告诉小 A，他的家位置于编号为 hi​ 的结点，并且他每秒钟能行走 1 米。请你帮小 A 计算，每位同学从家出发需要多少秒才能到达学校呢？

输入格式

第一行，四个正整数 n,m,s,q，分别表示 C 城的结点数与边数，学校所在的结点编号，以及小 A 同学们的数量。

接下来 m 行，每行三个正整数 ui​,vi​,li​，表示 C 城中的一条无向边。

接下来 q 行，每行一个正整数 hi​，表示一位同学的情况。

输出格式

共 q 行，对于每位同学，输出一个整数，表示从家出发到学校的最短时间。

输入输出样例

输入 #1复制

5 5 3 3
1 2 3
2 3 2
3 4 1
4 5 3
1 4 2
5
1
4

输出 #1复制

4
3
1

说明/提示

本题采用捆绑测试。

对于 20% 的测试点，保证 q=1。

对于另外 20% 的测试点，保证 1≤n≤500，1≤m≤500。

对于所有测试点，保证 1≤n≤2×105，1≤m≤2×105，1≤q≤2×105，1≤ui​,vi​,s,hi​≤n，1≤li​≤106。

代码实现：

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
typedef long long ll;
const ll INF = 1e18;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n,m,s,q;
    cin >> n >> m >> s >> q;
    vector<vector<pair<int,int>>> adj(n+1);
    for(int i=0;i<m;i++){
        int u,v,l; cin>>u>>v>>l;
        adj[u].emplace_back(v,l);
        adj[v].emplace_back(u,l);
    }
    vector<ll> dist(n+1,INF);
    priority_queue<pair<ll,int>,vector<pair<ll,int>>,greater<>> hp;
    dist[s]=0; hp.emplace(0,s);
    while(!hp.empty()){
        auto [d,u]=hp.top();hp.pop();
        if(d>dist[u])continue;
        for(auto [v,w]:adj[u]){
            if(dist[v]>d+w){
                dist[v]=d+w;
                hp.emplace(dist[v],v);
            }
        }
    }
    while(q--){
        int h;cin>>h;
        cout << dist[h] << '\n';
    }
    return 0;
}