树形DP学习笔记

P5658 [CSP-S2019] 括号树

题目分析

1. 给定一棵有 $n$ 个节点的树，点 1 是根节点；
2. 每个节点上有 1 个括号，可能为(或者)。

目标：对于每一个节点 $i$，从点 1 到点 $i$ 的路径形成的括号串，求其合法括号子串的数量 $k_i$，并输出 $i\times k_i$ 的异或和。

思路分析

1. 将树上问题转化为序列问题

状态：$d{p}_i$ 表示已 $i$ 结尾的合法子串的数量。
答案：$i\times d{p}_i$ 并求异或和。

• 当前字符为(，$d{p}_i\leftarrow 0$；
• 当前字符为)，此时可以匹配，令此时距离最近的左括号下标为 $u$，$d{p}_i\leftarrow d{p}_{u-1}+1$

2. 放到树上去

状态转移：令此时祖先节点最近的左括号下标为 $u$，$u$ 的父节点 $f{a}_u$ 为 $w$，那么 $d{p}_i\leftarrow d{p}_w+1$。

此时需要注意，路径之间有公共部分，于是在 dfs 结束之后要重新将弹出的左括号重新入栈。

3. 最终答案需要取前缀和再异或

代码展示

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 5e5 + 5;
int fa[N] , dp[N];
vector<int> G[N];
stack<int> stk;
int n;
string s;
void dfs(int cur){
	int tmp;
	bool flag = 0;
	dp[cur] = 0;
	if(s[cur] == '('){//当前是一个左括号，无法匹配 
		stk.push(cur);
	}
	else if(!stk.empty()){//是一个右括号，并且栈不为空 
		tmp = stk.top();
		stk.pop();
		flag = 1;//标记已经匹配成功 
		dp[cur] = dp[fa[tmp]] + 1;
	}
	for(int i = 0 ; i < G[cur].size() ; i++){
		dfs(G[cur][i]);
	}
	if(flag){//递归回溯时要把他放回去 
		stk.push(tmp);
	}
	else if(s[cur] == '('){
		stk.pop();//回溯弹出 
	}
	return ;
}
void get_sum(int cur){
	for(int i = 0 ; i < G[cur].size() ; i++){
		dp[G[cur][i]] += dp[cur];//将父节点的答案加到子结点上 
		get_sum(G[cur][i]);//求和 
	}
	return ;
}
signed main(){
	cin>>n>>s;
	s = '.' + s;
	for(int i = 2 ; i <= n ; i++){
		cin>>fa[i];
		G[fa[i]].push_back(i);
	}
	dfs(1);
	get_sum(1);
	int ans = 0;
	for(int i = 1 ; i <= n ; i++){
		ans = ans ^ (i * dp[i]);//答案异或，注意要开long long，不然会爆炸 
	}
	cout<<ans;
	return 0;
}