题解 [LuoguP4974] 毒瘤之神秘通道
题意
原题面有点乱,这里补充个简易题面:
- 一棵 m m m 个节点的树,初始边权为 0 0 0,根节点为 0 0 0;
- 有 n n n 个操作,第 i i i 个操作给出 a i , b i a_i,b_i ai,bi,把路径 a → 0 a\to0 a→0 上的边权值分别加上 b i b_i bi;
- 定义 u → v u\to v u→v 的路径长度为路径 u → v u\to v u→v 上的边权值和;
- 求出 0 → a i 0\to a_i 0→ai 的路径长度最小值,若有多个最小值,输出 a i a_i ai 最大的那一个。
题解
记 v a l i val_i vali 为 a i → f a a i a_i\to fa_{a_i} ai→faai 的边权。
首先读入每个节点 i i i 的父亲 f a i fa_i fai,将 v a l a i val_{a_i} valai 设为 b i b_i bi,记录每个节点的入度。
然后进行拓扑排序,排序过程中每遍历到一个节点 i i i,就把 v a l f a i val_{fa_i} valfai 加上 v a l i val_i vali 即可。因为离根节点越近的节点被遍历到的时间就越晚,可以保证所有到根节点的路径上节点的遍历时间单调递增。
这里就计算完成了一个数组 v a l i val_i vali,表示操作完成后 a i → f a a i a_i\to fa_{a_i} ai→faai 的边权。
接着重新建图,连边 f a i → i fa_i\to i fai→i,权值为 v a l i val_i vali,从 0 0 0 节点开始 dfs,即可求出每个节点到 0 0 0 的路径长度。挑出所有 a i a_i ai 计算即可。
//P4974
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
long long val[N], ans[N], res = 0x7fffffffffffffff;
int n, m, fa[N], ind[N], ist[N], pos;
vector<int> g[N];
void dfs(int x){
for(int i = 0; i < g[x].size(); ++ i){
int y = g[x][i];
ans[y] = ans[x] + val[y];
dfs(y);
}
}
int main(){
scanf("%d%d", &m, &n);
for(int i = 1, p; i <= m; ++ i){
scanf("%d", &p);
fa[i] = p;
++ind[p];
}
for(int i = 1, a, b; i <= n; ++ i){
scanf("%d%d", &a, &b);
ist[a] = 1;
val[a] += b;
}
//拓扑排序
queue<int> q;
for(int i = 0; i <= m; ++ i){
if(!ind[i]) q.push(i);
}
while(!q.empty()){
int t = q.front();
q.pop();
val[fa[t]] += val[t];
-- ind[fa[t]];
if(!ind[fa[t]]) q.push(fa[t]);
}
//dfs
for(int i = 1; i <= m; ++ i){
g[fa[i]].push_back(i);
}
dfs(0);
for(int i = 0; i <= m; ++ i){
if(ist[i] && ans[i] <= res) res = ans[i], pos = i;
}
printf("%d %lld\n", pos, res);
return 0;
}