P5658 括號樹
阿新 • • 發佈:2020-08-22
P5658 括號樹
NOIp2019
我是永遠不會忘記我那天在考場上傻瞪著題啥都不會的心理陰影的……
於是今天我克服心理陰影來寫這道題。
樹形結構
因為這是一個樹,所有優秀的性質這個題都有。並且題目僅僅是問從1開始到所有點的答案,所以我們就可以依靠樹的性質來做。
首先,對於一個節點,我們給它記錄幾個值:
- \(lst_i\)表示i的貢獻(只是i的貢獻,並不包括從根節點到i路徑上點的貢獻)
- 那麼我們發現這個lst是如何轉移的。考慮一條到i的路徑,如果i是後括號,那麼\(lst_i\)即為與之配對的前括號的父節點的\(lst\)+1。因為假設這個前括號的父節點同樣有一個已經匹配了的後括號,那麼我們勢必可以把當前的匹配和之前的匹配序列合併,當前的這個後括號的貢獻值,其實就等於前面那個後括號的貢獻值+1
- 那麼我們發現這個lst是如何轉移的。考慮一條到i的路徑,如果i是後括號,那麼\(lst_i\)即為與之配對的前括號的父節點的\(lst\)+1。因為假設這個前括號的父節點同樣有一個已經匹配了的後括號,那麼我們勢必可以把當前的匹配和之前的匹配序列合併,當前的這個後括號的貢獻值,其實就等於前面那個後括號的貢獻值+1
- \(sum_i\)表示從根節點到i的貢獻的和。那麼轉移就是\(sum_i=sum_{fa_i}+lst_i\)。
最後答案即為每個sum乘i的異或和。
然後按照樹dfs一遍即可……嗎?
dfs
對於前括號和後括號我們用一個棧維護即可。遇到前括號入棧,後括號出棧並用上述方式處理,如果空棧就跳過。這裡只需要注意一點:dfs之後需要將棧內修改的值重新放回去。
其餘就沒什麼了。
(做不出來還是我太蔡了)
記得開long long。
程式碼
#include<iostream> #include<cstdio> #include<cstring> #include<cctype> #include<algorithm> #include<stack> #define int long long using namespace std; inline int read(){ int x=0,w=0;char c=getchar(); while(!isdigit(c))w|=c=='-',c=getchar(); while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar(); return w?-x:x; } namespace star { const int maxn=5e5+10; int n,fa[maxn],ans,lst[maxn]; char s[maxn]; int ecnt,head[maxn],nxt[maxn],to[maxn],sum[maxn]; inline void addedge(int a,int b){ to[++ecnt]=b,nxt[ecnt]=head[a],head[a]=ecnt; } stack<int> st; void dfs(int x){ int tmp=0; if(s[x]=='(')st.push(x); else{ if(!st.empty()){ tmp=st.top(); st.pop(); lst[x]=lst[fa[tmp]]+1; } } sum[x]=sum[fa[x]]+lst[x]; for(int i=head[x];i;i=nxt[i])dfs(to[i]); if(tmp)st.push(tmp); else if(!st.empty())st.pop(); } inline void work(){ n=read(); scanf("%s",s+1); for(int a,i=2;i<=n;i++) a=read(),addedge(a,i),fa[i]=a; dfs(1); for(int i=1;i<=n;i++) ans^=(sum[i]*i); printf("%lld",ans); } } signed main(){ star::work(); return 0; }