树上最近点对【最短路】

题意：
给定一棵 n 个点（点编号为 1,2,⋯,n） 的树，标记出k 个不同的点 s1​,s2​,⋯,sk​ ，对于每个 si​ 求出剩下的标记点中哪个离 si​ 最近，输出距离。

注意：这不是一个满分做法，复杂度 O(nlog(n)) ，树形dp复杂度可以达到 O(n)。

解析:
新建一个点向所有标记点连一条权值为 0 的边，并作为源点跑 Dijkstra ，可以求出离点 u 最 近的标记点 f(u) 和它们之间的距离dis(u) 。若一条边 < u, v, w > 满足 f(u) ≠ f(v) ，那么 f(u) 经过这条边到 f(v) 的最短距离一定是 dis(u) + w + dis(v) ，找出所有满足条件的边，取最小值。

代码：

#include<bits/stdc++.h>
#define lson sign << 1, l, mid
#define rson sign << 1 | 1, mid + 1, r
#define lowbit(x) (x&(-x))
typedef long long ll;
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1e6 + 5;

int n, k, s[maxn], flag[maxn], pre[maxn];
int father[maxn], cost[maxn];
ll ans[maxn], dis[maxn];
struct node {
    int e, c, p;
} load[maxn << 2];
int head[maxn], sign;
void add_edge(int s, int e, int c) {
    load[++sign] = node{e, c, head[s]};
    head[s] = sign;
}
struct state {
    int e;
    ll c;
    bool friend operator<(state a, state b) {
        return a.c > b.c;
    }
};
void dij() {
    priority_queue<state>q;
    q.push(state{0, 0});
    while(!q.empty()) {
        state w = q.top();
        q.pop();
        int s = w.e;
        for(int i = head[s], e; i; i = load[i].p) {
            e = load[i].e;
            if(dis[e] > w.c + load[i].c) {
                dis[e] = w.c + load[i].c;
                if(flag[e])
                    pre[e] = e;
                else
                    pre[e] = pre[s];
                q.push(state{e, dis[e]});
            }
        }
    }
}
void init() {
    memset(ans, 0x3f, sizeof(ans));
    memset(dis, 0x3f, sizeof(dis));
    dis[0] = 0;
}
int main() {
    init();
    scanf("%d", &n);
    for(int i = 2; i <= n; i++) {
        scanf("%d %d", &father[i], &cost[i]);
        add_edge(father[i], i, cost[i]);
        add_edge(i, father[i], cost[i]);
    }
    scanf("%d", &k);
    for(int i = 1; i <= k; i++) {
        scanf("%d", &s[i]);
        flag[s[i]] = 1;
        add_edge(0, s[i], 0);
        add_edge(s[i], 0, 0);
    }
    dij();
    for(int i = 2; i <= n; i++) {
        int s = i, e = father[i];
        if(pre[s] == pre[e])
            continue;
        ll w = dis[e] - dis[pre[e]] + cost[i] + dis[s] - dis[pre[s]];
        ans[pre[s]] = min(ans[pre[s]], w);
        ans[pre[e]] = min(ans[pre[e]], w);
    }
    for(int i = 1; i <= k; i++)
        printf("%lld ", ans[s[i]]);
    return 0;
}