2022杭电多校第二场 A.Static Query on Tree(树剖)

#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
//#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define yes cout<<"Yes\n"
#define no cout<<"No\n"
#define yesno if(fg)yes;else no;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=8e18;
const int maxn=2e5+100;
vector<int>g[maxn];
int a[maxn],b[maxn],c[maxn],son[maxn],siz[maxn],fa[maxn];
int dep[maxn],top[maxn],dfn[maxn],rk[maxn],clk,out[maxn];
template <typename T> inline void read(T& x)
{
	x=0;
	T f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-')f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	x=x*f;
	return;
}
template <typename T,typename ...Arg>void read(T& x,Arg& ...arg)
{
	read(x);
	read(arg...);
}
template <typename T>inline void write(T x)
{
	if(x<0)putchar('-'),x=-x;
	if(x<10)putchar(x+'0');
	else write(x/10),putchar(x%10+'0');
}
template <typename T,typename ...Arg>void write(T& x,Arg& ...arg)
{
	write(x);
	putchar(' ');
	write(arg...);
}

struct node
{
	int l,r;
	int sum;
	int s[8];
	int lazy;
	int lazy2;//清空标志
} t[maxn*4];
void dfs(int x,int f)
{
	fa[x]=f;
	dep[x]=dep[f]+1;
	siz[x]=1;
	son[x]=-1;
	for(auto it:g[x])
	{
		if(it==f)continue;
		dfs(it,x);
		siz[x]+=siz[it];
		if(son[x]==-1||siz[it]>siz[son[x]])son[x]=it;
	}
}
void dfs2(int x,int t)
{
	top[x]=t;
	dfn[x]=++clk;
	rk[clk]=x;
	if(son[x]==-1)
	{
		out[x]=clk;
		return ;
	}
	dfs2(son[x],t);
	for(auto it:g[x])
	{
		if(it==fa[x])continue;
		if(it==son[x])continue;
		dfs2(it,it);
	}
	out[x]=clk;
}
void push_down(int k)
{
	if(t[k].l==t[k].r)
	{
		t[k].lazy=0;
		t[k].lazy2=0;
		return ;
	}
	if(t[k].lazy2)//清空 
	{
		t[k<<1].lazy2=t[k<<1|1].lazy2=1;
		t[k<<1].lazy=t[k<<1|1].lazy=0;//note
		for(int i=0; i<=7; i++)
		{
			t[k<<1].s[i]=t[k<<1|1].s[i]=0;
		}

		t[k<<1].s[0]=t[k<<1].r-t[k<<1].l+1;
		t[k<<1|1].s[0]=t[k<<1|1].r-t[k<<1|1].l+1;

		t[k].lazy2=0;
	}
	if(t[k].lazy>0)
	{
		for(int i=7; i>=0; i--)
		{
			if((i|t[k].lazy)==i)continue;

			t[k<<1].s[i|t[k].lazy]+=t[k<<1].s[i];
			t[k<<1].s[i]=0;

			t[k<<1|1].s[i|t[k].lazy]+=t[k<<1|1].s[i];
			t[k<<1|1].s[i]=0;
		}
		t[k<<1].lazy|=t[k].lazy;
		t[k<<1|1].lazy|=t[k].lazy;
		t[k].lazy=0;
	}



}
void push_up(int k)
{
	if(t[k].l!=t[k].r)
	{
		for(int i=0; i<8; i++)
		{
			t[k].s[i]=t[k<<1].s[i]+t[k<<1|1].s[i];
		}
	}
}

void build(int l,int r,int k)
{
	t[k].l=l;
	t[k].r=r;
	t[k].sum=0;
	for(int i=0; i<8; i++)t[k].s[i]=0;
	t[k].s[0]=t[k].r-t[k].l+1;
	t[k].lazy=0;
	t[k].lazy2=0;
	if(l==r)
	{
		return ;
	}
	int mid=(l+r)>>1;
	build(l,mid,k<<1);
	build(mid+1,r,k<<1|1);
	push_up(k);
}
void update(int l,int r,int k,int type,int val)
{
	if(r<t[k].l||l>t[k].r)return ;
	push_down(k);
	if(l<=t[k].l&&t[k].r<=r)
	{
		if(val==1)
		{
			t[k].lazy=type;
			for(int i=7; i>=0; i--)
			{
				if((i|type)==i)continue;
				t[k].s[i|type]+=t[k].s[i];
				t[k].s[i]=0;
			}
		}
		else
		{
			t[k].lazy2=1;
			for(int i=0; i<8; i++)t[k].s[i]=0;
			t[k].s[0]=t[k].r-t[k].l+1;
		}
		push_down(k);
		return ;
	}
	update(l,r,k<<1,type,val);
	update(l,r,k<<1|1,type,val);
	push_up(k);
}

int querysum(int l,int r,int k)
{
	if(r<t[k].l||l>t[k].r)return 0;
	push_down(k);
	if(l<=t[k].l&&t[k].r<=r)
	{
		return t[k].s[7];
	}
	int ans=0;
	ans+=querysum(l,r,k<<1);
	ans+=querysum(l,r,k<<1|1);
	return ans;
}
int na,nb,nc;
int n,q;
signed main()
{
	int tt;
	read(tt);
	int cas=0;
	while(tt--)
	{
		cas++;
		clk=0;
		read(n,q);
		for(int i=2; i<=n; i++)
		{
			int x;
			read(x);
			g[x].pb(i);
			g[i].pb(x);
		}
		dfs(1,0);
		dfs2(1,0);
		build(1,n,1);

		while(q--)
		{
			read(na,nb,nc);
			for(int i=1; i<=na; i++)
			{
				read(a[i]);
				int x=1,y=a[i];
				while(top[x]!=top[y])
				{
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					update(dfn[top[x]],dfn[x],1,1,1);
					x=fa[top[x]];
				}
				if(dfn[x]<dfn[y]) swap(x,y);
				update(dfn[y],dfn[x],1,1,1);
			}
			for(int i=1; i<=nb; i++)
			{
				read(b[i]);
				int x=1,y=b[i];
				while(top[x]!=top[y])
				{
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					update(dfn[top[x]],dfn[x],1,2,1);
					x=fa[top[x]];
				}
				if(dfn[x]<dfn[y]) swap(x,y);
				update(dfn[y],dfn[x],1,2,1);
			}
			for(int i=1; i<=nc; i++)
			{
				read(c[i]);
				int x=c[i];
				update(dfn[x],out[x],1,4,1);
			}
			int ans=querysum(1,out[1],1);
			write(ans);
			printf("\n");

			update(1,out[1],1,1,-1);

		}
		build(1,n,1);
		for(int i=0; i<=n+2; i++)
		{
			g[i].clear();
			a[i]=b[i]=c[i]=son[i]=siz[i]=fa[i]=0;
			dep[i]=top[i]=dfn[i]=rk[i]=clk=out[i]=0;
		}

	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

thusloop

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值