codeforces 1044F DFS dfs序+线段树

题目链接

题意：给你一棵树，然后有q次操作，每次操作在x，y节点连一条边，如果之前有边，就删除，但是不删除原来树的边，输出这颗树有多少个好节点，好节点的定义是，以这个节点为根节点，dfs这棵树，如果存在一种情况，dfs经过的边都是原来树的边，那么这个节点就是好节点。

思路：先偷个图,嘻嘻。

思路是，我们求出不好的点，答案就是总共的点减去不好的点，分为两种情况，情况1是y节点位于 x 节点的子树上，这时，x，y所夹的这段区间上的点都是不好的点，情况二是x，y不在一棵子树上，那么，此时x，y上方的点都是不好的点。怎么证明？你想，如果以一个节点为根节点，他的不同的子树上有边，那么，那这个节点必然是不好，多换换角度看看这颗树，也许就能想明白这个。那么我们求出这棵树的dfs序，然后，根据上图的两种不同的类型分别修改区间，注意区间修改的小技巧，如果这个区间标记为1，那么他的值就等于区间长度，否则等于左儿子加右儿子，如果是根节点直接赋为0，避免出现 runtime  error 这种情况。

 

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
vector<int>G[maxn];
map<int,int>mp[maxn];
int dep[maxn],f[maxn][20],l[maxn],r[maxn];
int sum[maxn*4],lz[maxn*4],cnt=0;

#define ls o<<1
#define rs o<<1|1
#define mid (l+r)/2

void dfs(int u,int fa,int deep)
{
	dep[u]=deep;
	f[u][0]=fa;
	l[u]=++cnt;
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v==fa)continue;
		dfs(v,u,deep+1);
	}
	r[u]=cnt;
}
void up(int o,int l,int r,int L,int R,int v)
{
	if(l>r)return ;
	if(l>=L&&r<=R)
	{
		lz[o]+=v;
		if(lz[o])sum[o]=r-l+1;
		else if(l==r)sum[o]=0;
		else sum[o]=sum[ls]+sum[rs];
		return ;
	}
	if(L<=mid)up(ls,l,mid,L,R,v);
	if(R>mid)up(rs,mid+1,r,L,R,v);
	if(lz[o])sum[o]=r-l+1;
	else if(l==r)sum[o]=0;
	else sum[o]=sum[ls]+sum[rs];
}
int main()
{
	int n,q,u,v,w;
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0,1);
	for(int j=1;j<20;j++)
	{
		for(int i=1;i<=n;i++)
		{
			f[i][j]=f[f[i][j-1]][j-1];
		}
	}
	while(q--)
	{
		scanf("%d%d",&u,&v);
		if(l[u]<l[v])swap(u,v);
		if(!mp[u][v])mp[u][v]=1,w=1;
		else mp[u][v]=0,w=-1;
		if(r[u]<=r[v])
		{
			int U=u;
			for(int i=19;i>=0;i--)
			{
				if(dep[f[u][i]]>dep[v])u=f[u][i];
			}
			up(1,1,n,l[u],l[U]-1,w);
			up(1,1,n,r[U]+1,r[u],w);
		}
		else 
		{
			up(1,1,n,1,l[v]-1,w);
			up(1,1,n,r[v]+1,l[u]-1,w);
			up(1,1,n,r[u]+1,n,w);
		}
		printf("%d\n",n-sum[1]);
	}
}