P8844 [傳智杯 #4 初賽] 小卡與落葉 題解

harmis_yz發表於2024-03-05

分析

亂搞題。

$1 \le n,m \le 10^5$ 的時候就可以考慮亂搞了。

發現每次操作 $1$ 都會把上一次的操作 $1$ 覆蓋掉,那麼第 $i$ 個詢問時樹的顏色情況就是由前 $1$ 個操作 $1$ 決定。也就是說這個詢問的內容變成了:在 $x$ 為根的子樹中,深度不小於 $x'$ 的節點數量。$x'$ 是該操作 $1$ 的 $x$。

求這玩意直接樹上啟發式合併。記錄答案的時候用值域分塊求一下和就行。複雜度 $O(m \sqrt{n})$。

程式碼

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define re register
#define il inline
#define PII pair<int,int>
#define x first
#define y second

const int N=1e5+10;
int n,m;
int ne[N<<1],e[N<<1],h[N],idx;
int dep[N],mson[N],siz[N];
int in[N],out[N],dfsx[N],cnt;
int cnt_dep[N];
int sum[N],len;
vector<PII> Q[N];
int ans[N],qidx;

il void add(int a,int b){ne[++idx]=h[a],e[idx]=b,h[a]=idx;}
il int get(int x){return (x-1)/len+1;}
il void Add(int x){++cnt_dep[dep[x]],++sum[get(dep[x])];}
il void Del(int x){--cnt_dep[dep[x]],--sum[get(dep[x])];}
il void merge(int x,int flag){
	for(re int i=in[x];i<=out[x];++i)
		if(flag) Add(dfsx[i]);
		else Del(dfsx[i]);
	return ;
}
il int query(int x){
	int l=x,r=n,bl=get(x),br=get(n),ans=0;
	if(bl==br){
		for(re int i=l;i<=r;++i) ans+=cnt_dep[i];
		return ans;
	}
	for(re int i=l;i<=bl*len;++i) ans+=cnt_dep[i];
	for(re int bk=bl+1;bk<=br-1;++bk) ans+=sum[bk];
	for(re int i=(br-1)*len+1;i<=r;++i) ans+=cnt_dep[i];
	return ans;
}
il void dfs1(int now,int fa){
	dfsx[++cnt]=now,in[now]=cnt;
	dep[now]=dep[fa]+1,mson[now]=-1,siz[now]=1;
	int msiz=0;
	for(re int i=h[now];i;i=ne[i]){
		int j=e[i];if(j==fa) continue;
		dfs1(j,now),siz[now]+=siz[j];
		if(msiz<siz[j]) msiz=siz[j],mson[now]=j;	
	}
	out[now]=cnt;
	return ;
}
il void dfs2(int now,int fa,int flag){
	for(re int i=h[now];i;i=ne[i]){
		int j=e[i];if(j==fa||j==mson[now]) continue;
		dfs2(j,now,1);
	}
	if(mson[now]!=-1) dfs2(mson[now],now,0);
	for(re int i=h[now];i;i=ne[i]){
		int j=e[i];if(j==fa||j==mson[now]) continue;
		merge(j,1);
	}
	Add(now);
	for(re int j=0;j<Q[now].size();++j) ans[Q[now][j].x]=query(Q[now][j].y);
	if(flag) merge(now,0);
	return ;
}

il void solve(){
	cin>>n>>m;len=sqrt(n);
	for(re int i=1,a,b;i<n;++i) 
		cin>>a>>b,add(a,b),add(b,a);
	int min_y=n+1;
	for(re int i=1;i<=m;++i){
		int op,x;cin>>op>>x;
		if(op==1) min_y=x;
		else Q[x].push_back({++qidx,min_y});
	}
	dfs1(1,0),dfs2(1,0,0);
	for(re int i=1;i<=qidx;++i) cout<<ans[i]<<"\n";
}

signed main(){
	solve();
	return 0;
}

相關文章