E. Trees of Tranquillity(括号化序列)
题目大意:
给你两课树,找个点集使在第一棵树上两两之间有祖先关系,第二课树上两两之间没有祖先关系。
思路:
在第二颗树上遍dfs,即可获得括号化序列
括号化序列:
括号化序列为,只有不相交和包含关系的区间。被包含的节点在该节点子树里。
即有祖先关系。
我们要在第一棵树上跑 d f s dfs dfs,按照 d f s dfs dfs顺序维护区间即可保证集合在第一棵树上两两之间有祖先关系,且区间不相交,即保证在第二棵树上两两之间没有祖先关系。
我们将区间加入到 s e t set set里面,每次加区间是,二分其左端点,如果其左端点与 s e t set set里面区间相交,即与其相交的区间必定包含该区间,我们贪心的把大区间更新掉。
这个题的巧妙之处在于输入 a i < i a_i<i ai<i这个限制,这样一条链上的各个点标号一定是递增的。第二棵树上因为标号大的区间一定不会覆盖标号小的区间这一性质,所以可以这样贪心。
代码:
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#define fi first
#define se second
using namespace std;
typedef pair <int,int> pll;
const int N=3e5+7;
int T,n,Ans,tim,in[N],out[N];
int a[N],b[N];
vector <int> G1[N],G2[N];
set <pll> sa;
void init(){
sa.clear();
for(int i=1;i<=n;i++){
G1[i].clear();
G2[i].clear();
in[i]=out[i]=0;
}
tim=Ans=0;
}
void dfs2(int p){
in[p]=++tim;
for(int i=0;i<G2[p].size();i++){
int v=G2[p][i];
dfs2(v);
}
out[p]=tim;
}
void dfs1(int p){
int add=0,del=0;
auto it=sa.lower_bound(pll(in[p],out[p]));
pll t;
if(it!=sa.begin()){
it--;
if(out[p]<=(*it).se){
del=1;
t=*it;
sa.erase(it);
}
}
add=1;
sa.insert(pll(in[p],out[p]));
Ans=max(Ans,(int)sa.size());
for(int i=0;i<G1[p].size();i++){
int v=G1[p][i];
dfs1(v);
}
if(add) sa.erase(pll(in[p],out[p]));
if(del) sa.insert(t);
}
void solve(){
cin>>n;
init();
for(int i=2;i<=n;i++){
int fa;
cin>>fa;
G1[fa].push_back(i);
}
for(int i=2;i<=n;i++){
int fa;
cin>>fa;
G2[fa].push_back(i);
}
dfs2(1);
dfs1(1);
cout<<Ans<<"\n";
}
int main(){
cin>>T;
while(T--) solve();
}