Codeforces Round #800 (Div. 2) D. Fake Plastic Trees

link
拓扑排序

题意

给你一棵以1为根的树，你可以进行以下操作若干次：
随意选中一个点，设从根到这个点的简单路径一共有 $n$ 个点，则你可以任选一个序列 $c_1,c_2,...,c_n(0\le c_1\le ...\le c_n)$ ，从根开始依次将这 $n$ 个点的权值加上 $c_i$。
现给出每个点的 $l_i,r_i$，问至少多少次操作可以使得所有点的权值都处在 $[l_i,r_i]$ 内。

思路

设 $val[i]$ 表示点 $i$ 可以被加的最大权值，初始均为0.
首先一个显而易见的结论： $ans$ 最多为 $n$ 次。因为对每个点都可以前面的点都取零，最后一个点取 $r_i$ 。
于是我们从叶节点开始考虑：由上面的结论，我们肯定是希望叶节点一次被取掉的，这里我们直接把叶节点的 $val[i]=r[i]$ ，并累加一次ans（因为序列是单增的，相比取 $l[i]$, $r[i]$ 显然更优）。然后每个父节点的 $val[fa]$ 便是它的子节点的 $val[son]$ 之和，并根据 $val[fa]$ 和$l[fa]$ 之间的关系维护 $ans$。用拓扑排序实现遍历即可。

注意两个点：

1. 如果叶节点的 $l[i]=0$ ，则直接 $val[i]=0$ 即可，因为这个点相当于不存在。
2. 在更新 $val[fa]$ 时要注意在已经遍历完所有子节点后，若 $val[fa]<l[fa]$，则需要一次操作（相当于从1走到 $fa$ 节点，$c_i=r[fa]-val[fa]$），并令 $val[fa]=r[fa]$，若 $val[fa]>r[fa]$ ，则要令 $val[fa]=r[fa]$。 原因是不能超过上限，这里 $val[fa]$ 即便减少了也不会影响已经遍历的节点（即它的子节点），因为序列是单调增的，前面的项变小了不会影响单调性。

代码

int n;
int l[maxn], r[maxn];
int val[maxn];
int pa[maxn], in[maxn];
int ans = 0;
int toposort() {
	queue<int> q;
	for(int i = 1; i <= n; i++) {
		if(!in[i]) {
			q.push(i);
			if(l[i] > 0) {
				val[i] = r[i];
				ans++;
			}
			else {
				val[i] = 0;
			}
		}
	}
	while(!q.empty()) {
		int p = q.front();
		q.pop();
		int fa = pa[p];
		in[fa]--;
		val[fa] += val[p];
		if(in[fa] == 0) {
			q.push(fa);
			if(val[fa] < l[fa]) {
				val[fa] = r[fa];
				ans++;
			}
			val[fa] = min(val[fa], r[fa]);
		}
	}
	return ans;
}
void solve() {
    cin >> n;
    ans = 0;
    pa[1] = 1;
    for(int i = 1; i <= n; i++) {
    	in[i] = 0;
    }
    for(int i = 2; i <= n; i++) {
    	int x;
    	cin >> x;
    	in[x]++;
        pa[i] = x;
    }
    for(int i = 1; i <= n; i++) {
    	cin >> l[i] >> r[i];
    	val[i] = 0;
    }
    cout << toposort() << endl;
}