E. Blood Cousins Return (树上启发式合并维护set)

传送门

给定了一个森林,每棵树都是有根树,m次询问,每次询问节点x的k级后代中,有多少个互不相同的名字。

对于第i层,维护一个set表示这层的名字集合,达到去重效果。

把查询离线,然后dsu on tree即可。

#include<bits/stdc++.h>
using namespace std;
//#pragma GCC optimize(2)
#define ull unsigned long long
#define ll long long
#define pii pair<int, int>
const int maxn = 2e5 + 10;
const ll mod = 1e9+7;
const ll inf = (ll)4e17+5;
const int INF = 1e9 + 7;
const double pi = acos(-1.0);
ll inv(ll b){
    
    if(b==1)return 1;return(mod-mod/b)*inv(mod%b)%mod;}
//给定森林 m次询问 每次问 x的k级后代中 名字不同的人的个数
string name[maxn];
int n,m;
vector<int> g[maxn];
set<string> ans[maxn];//每层的名字
int fa[maxn];
vector<pii> q[maxn];
int size[maxn],son[maxn],dep[maxn];
int ret[maxn];
void dfs(int rt,int fa) 
{
    
    
	dep[rt]=dep[fa]+1;
	size[rt]=1;
	for(int i:g[rt]) 
	{
    
    
		dfs(i,rt);
		size[rt]+=size[i];
		if(size[i] > size[son[rt]]) son[rt]=i;
	}
}
int SON;
void add(int rt,int v) 
{
    
    
	if(v)
		ans[dep[rt]].insert(name[rt]);
	else ans[dep[rt]].erase(name[rt]);
	for(int i:g[rt]) 
	{
    
    
		if(i==SON) continue;//重儿子保留
		add(i,v);
	}
}
 
void dfs2(int rt,bool ok) 
{
    
    
	for(int i:g[rt]) 
	{
    
    
		if(i==son[rt]) continue;
		dfs2(i,0);
	}
	if(son[rt]) dfs2(son[rt],1),SON=son[rt];
	add(rt,1),SON=0;
	for(auto i:q[rt]) //rt子树中  深度为dep[rt]+k的 set的size
	{
    
    
		ret[i.second]=ans[dep[rt]+i.first].size();
	}
	if(!ok) add(rt,0);
}
int main()
{
    
    
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
    
    
		cin>>name[i]>>fa[i];
		g[fa[i]].push_back(i);
	}
	for(int i:g[0]) dfs(i,0);
	scanf("%d",&m);
	for(int i=1;i<=m;i++) 
	{
    
    
		int  x,k;
		scanf("%d %d",&x,&k);
		q[x].push_back({
    
    k,i});
	}
	for(int i:g[0]) dfs2(i,0);
	for(int i=1;i<=m;i++) cout<<ret[i]<<'\n';
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_46030630/article/details/120588587