2021牛客暑期多校训练营7 F题: xay loves trees

F题: xay loves trees

原题链接:https://ac.nowcoder.com/acm/contest/11258/F

题目大意

给定两个 $n$ 个节点且根为 $1$ 的树，求 { 1 , 2 , . . . , n } \{1,2,...,n\} {1,2,...,n} 的最大子集满足:

1. 在第一棵树中，子集中的点相互连通，且对于任意两点 $u,v$ ，其中一个点必定是另一个点的祖先;
2. 在第二棵树中，对于任意两点 $u,v$ ， $u$ 不是 $v$ 的祖先， $v$ 也不是 $u$ 的祖先;

求最大子集的大小。

题解

在完成本题之前，建议了解原题CF1528C/1529E(当然你不想了解也行，尽量做到全面，不会省略原题的解题步骤)
链接:https://codeforces.com/problemset/problem/1528/C
本题与原题的区别:在第一棵树中，子集中的点相互连通。

本题有两个条件需要达成，我们考虑先满足其中的一个，然后快速地检查另一项是否成立。
显然，我们对第二棵树进行枚举情况的难度很大(第一棵树中的点显然形成一条链，下文会讲搜索方法)，那么我们考虑如何检查第二棵树中节点是否合法，这可以用dfs序实现，如下例:

我们对于每一个点记录下其dfs递归进入和递归返回时的dfs序，显然，若两个点的dfs序组成的区间有重合，则它们肯定呈祖先关系，反之，两点肯定不为祖先关系。
对于区间是否重合的维护，我们可以使用线段树，每次加入一个区间时，将其所在的区间染色，然后判断是否有重色即可。

然后考虑在第一棵树中搜索，因为联通且都存在祖先关系，所以一个合法点集在第一棵树中肯定呈现为一条从上往下的链，那么我们可以从根节点开始往下搜索，用尺取法(树上滑动窗口)的思想完成这个问题:每次我们放入一个点，然后判断当前是否是一种合法方案，若是则直接将当前点集大小加入统计，否则从上往下删除节点直至合法再加入统计。
但上述方案有一定的弊端，我们可能在dfs过程中在同一个点多次实现 非法-从上往下删除节点至合法-递归-回溯 的过程，若为类菊花图则可能导致时间复杂度过大，因此我们加入以下优化:每次出现非法情况时，我们只从链的上端删除一个点，然后判断是否合法，若是则将当前点集大小加入统计。
证明正确性:若当前链出现非法，则新点集大小=原点集大小+1(加入新节点)-1(删除上端点)=原点集大小，即新出现的非法情况的点集大小与原合法点集的大小是相同的，当最优解(即点集最大大小)更新时，肯定在某处形成了合法解，因为只有合法解省去了-1(删除上端点)一步，才可能使得点集大小增加。

参考代码

#include<bits/stdc++.h>
#define ll long long
#define For(i,n,m) for(int i=n;i<=m;i++)
#define FOR(i,n,m) for(int i=n;i>=m;i--)
#define pb push_back
#define mp make_pair
using namespace std;
void read(int &x){
	int ret=0;
	char c=getchar(),last=' ';
	while(!isdigit(c))last=c,c=getchar();
	while(isdigit(c))ret=ret*10+c-'0',c=getchar();
	x=last=='-'?-ret:ret;
}

const int MAXN=3e5+5;
struct node{
	int l,r,f,inc;
}segtree[MAXN<<2];

void pushup(int x){segtree[x].f=max(segtree[x<<1].f,segtree[x<<1|1].f);}//取较大者返回,我们需要的是判断是否有重复染色
void pushdown(int x){//将懒惰标记往下传递
	segtree[x<<1].f+=segtree[x].inc;
	segtree[x<<1|1].f+=segtree[x].inc;
	segtree[x<<1].inc+=segtree[x].inc;
	segtree[x<<1|1].inc+=segtree[x].inc;
	segtree[x].inc=0;//清空
}
void build(int x,int l,int r){//建树
	segtree[x].l=l,segtree[x].r=r,segtree[x].f=segtree[x].inc=0;//初始化
	if(l==r)return;
	int mid=l+r>>1;
	build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
}
void update(int x,int l,int r,int inc){
	if(segtree[x].l==l&&segtree[x].r==r){
		segtree[x].f+=inc;
		segtree[x].inc+=inc;//标记
		return;
	}
	if(segtree[x].inc)pushdown(x);//下传标记
	int mid=segtree[x].l+segtree[x].r>>1;
	if(r<=mid)update(x<<1,l,r,inc);
	else if(l>mid)update(x<<1|1,l,r,inc);
	else update(x<<1,l,mid,inc),update(x<<1|1,mid+1,r,inc);
	pushup(x);//更新
}

int query(int x,int l,int r){//询问
	if(segtree[x].l==l&&segtree[x].r==r){
		return segtree[x].f;
	}
	if(segtree[x].inc)pushdown(x);//下传标记
	int mid=segtree[x].l+segtree[x].r>>1;
	if(r<=mid)return query(x<<1,l,r);
	else if(l>mid)return query(x<<1|1,l,r);
	else return max(query(x<<1,l,mid),query(x<<1|1,mid+1,r));
}

int t,n,dfn[MAXN],edn[MAXN],cnt,way[MAXN],vis[MAXN],dp[MAXN];
vector<int>e1[MAXN],e2[MAXN];

void dfsn(int x,int fa){//跑出dfs序
	cnt++;
	dfn[x]=cnt;//进入
	int son;
	For(i,0,e2[x].size()-1){
		son=e2[x][i];
		if(son==fa)continue;
		dfsn(son,x);
	}
	edn[x]=cnt;//返回
}

void dfs(int x,int fa,int anc){//anc记录链顶祖先
	update(1,dfn[x],edn[x],1);//先加入该点
	int tmp=query(1,1,n);//询问是否有重复染色
	if(tmp>1){//非法
		dp[x]=dp[fa];
		vis[x]=1;//非法标记
		update(1,dfn[anc],edn[anc],-1);//将顶端删去
		int son;
		For(i,0,e1[x].size()-1){
			son=e1[x][i];
			if(son==fa)continue;
			way[x]=son;//记录当前点选择的子节点
			dfs(son,x,way[anc]);//anc改为anc所选择的子节点
		}
		update(1,dfn[x],edn[x],-1);//回溯
		update(1,dfn[anc],edn[anc],1);//回溯
	}
	else{
		dp[x]=dp[fa]+1;//子集数量增大
		int son;
		For(i,0,e1[x].size()-1){
			son=e1[x][i];
			if(son==fa)continue;
			way[x]=son;//记录当前点选择的子节点
			dfs(son,x,anc);//链顶端仍为anc
		}
		update(1,dfn[x],edn[x],-1);//回溯
	}
}

int main()
{
	//std::ios::sync_with_stdio(false),cin.tie(0);
	read(t);
	way[0]=1;
	while(t--){
		read(n);
		For(i,1,n)e1[i].clear(),e2[i].clear();
		memset(vis,0,sizeof(vis));
		memset(dp,0,sizeof(dp));
		int u,v;
		For(i,2,n){
			read(u),read(v);
			e1[u].pb(v);
			e1[v].pb(u);
		}
		For(i,2,n){
			read(u),read(v);
			e2[u].pb(v);
			e2[v].pb(u);
		}
		build(1,1,n);//建树
		cnt=0;
		dfsn(1,0);//跑出dfn与edn
		dfs(1,0,1);//尺取求解
		int ans=0;
		For(i,1,n)if(!vis[i])ans=max(ans,dp[i]);//取最大(合法)解
		printf("%d\n",ans);
	}
	return 0;
}