[补题] Maximize the Root （cf div.2）

D.Maximize the Root

思路:
题意是一颗树，可以将某个结点值+1，同时以这个结点为根的子树上的每个结点都-1(根节点除外)。不能有结点<0，可以无限次操作，求如何操作使根结点(1号结点)值最大。
注意到，根结点可以提高的最大值就是除了1以外所有结点的最小值。所以dfs一下，对于中间某一个结点x来说，如果它的子树中的最小值minN小于x，那么无法操作；如果子树中的最小值minN大于a[x]，则考虑通过操作使得a[x]提高直到minN<=a[x]。
一个简单的dfs。当时卡了很久结果tle了，赛后加以改进使用dp的思路来保存最小值，在dfs的过程中用b[x]存储以x为根的子树中的最小值。

代码:

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
void solve();
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int T = 1;
	cin >> T;
	while (T--) {
		solve();
	}

	return 0;
}

int n;
int a[200005],b[200005];
int p[200005];
vector<int> s[200005];

void dfs(int x) {  //dfs递归搜索从根节点开始
	int mn=1e9+1;   //minN是以x为根的子树的最小值
	for (int e : s[x]) {
		dfs(e);      //先向下搜索
		mn = min(mn,b[e]);  //下一层搜索完后得到的minN存在b[e]中，再一层层往回dp得到新的minN
	}
	if(mn == 1e9+1){  //叶子节点的minN就是自身
		b[x]=a[x];
		return;
	}
	if(x==1){  //根节点保存答案到a[1]，结束dfs
		a[1]+=mn;
		return;
	}
	if(a[x]<mn){ //中间节点，判断是否有必要操作来提高a[x]的值（比较a[x]和子树中的最小值，即minN）
		b[x] =(a[x]+mn)/2; //将儿子结点的值往上提，直到其minN<=a[x]，新的minN是(a[x]+mn)/2，存在b[x]中
	}
	else b[x]=mn;
}

void solve() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		b[i]=0;
		s[i].clear();
	}
	for (int i = 2; i <= n; i++) {
		cin >> p[i];
		s[p[i]].push_back(i);
	}
	dfs(1);		
	cout << a[1] << endl;
}