[被踩计划] 题解 括号树
为什么叫被踩记录呢?因为感觉自己之前真的是太菜了,打算把之前联赛等考过的题目做一做,看看自已以前有多菜,所以取名叫被踩记录。
题目分析
先考虑在一个序列上面做,然后再扩展到树上。
首先,我们先计算以每个位置作为结尾的合法字串个数,然后再通过后缀和统计答案。
如何求以某个位置作为结尾的合法字串个数?我们可以对每个位置找到长度最短的一个合法后缀,然后就转化成了一个子问题,可以用dp
做了。
举个例子:对于括号序列)())()(())
,以最后一个位置作为结尾的合法后缀就是括号序列)())()
以最后一个位置作为结尾的合法后缀数量加一,因为)())()(())
=)())()
+(())
,而(())
是一个合法括号串。
如何求出这个长度最短的一个合法后缀?可以用栈来解决这个问题,栈中维护的大概是这样一个序列))...))((...((
,同时维护栈中每个括号对应的位置,再分类讨论一下就行了,每次操作都是 \(\mathcal O(1)\) 的。
扩展到树上,做法还是一样的,可以再dfs
的同时顺便维护栈,只要回退的时候恢复就行了。
当然也可以不用dfs
,考虑到每次弹栈然后入栈类似一个树的结构,所以只需要维护在这个“栈树”上的父亲即可实现弹栈和入栈了。
参考代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=500005;
char s[maxn];
int pa[maxn],top[maxn],fa[maxn];
long long cnt[maxn];
int main(){
int n;
scanf("%d%s",&n,s+1);
pa[1]=0,top[1]=1;
for(int i=2;i<=n;++i){
int f;scanf("%d",&f);fa[i]=f;
if(s[i]=='('||!top[f]||s[top[f]]==')')
pa[top[i]=i]=top[f];
else
top[i]=pa[top[f]],cnt[i]=cnt[fa[top[f]]]+1;
}
for(int i=2;i<=n;++i)cnt[i]+=cnt[fa[i]];
long long ans=0;
for(int i=1;i<=n;++i)ans^=cnt[i]*i;
printf("%lld\n",ans);
return 0;
}