题意:
给定一棵 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;
}