CF-620-E-DFS序+线段树

合集 - Codeforces(31)

1.CF-55-D-数位DP2024-01-232.CF-87-D-并查集2024-01-133.CF-91-B-单调栈+二分2024-01-234.CF-240-F-线段树2024-01-215.CF-242-E-线段树2024-01-206.CF-292-D-并查集2024-01-237.CF-282-E-Trie2024-01-178.CF-342-E-分块2024-01-129.CF-375-D-启发式合并2024-01-1210.CF-431-D-二分+数位DP2024-01-2411.CF-461-B-树形DP2024-01-2212.CF-514-D-单调队列2024-01-1413.CF-522-D-线段树2024-05-0514.CF-570-D-启发式合并2024-01-2115.CF-600-E-启发式合并2024-05-0416.CF-613-D-虚树2024-01-13

17.CF-620-E-DFS序+线段树2024-01-17

18.CF-675-E-RMQ优化DP2024-01-1619.CF-702-E-倍增2024-01-1720.CF-797-E-根号分治2024-05-0421.CF-817-E-Trie2024-01-1622.CF-877-E-dfs序+线段树2024-05-0523.CF-915-F-并查集2024-01-2124.CF-938-D-最短路2024-05-1025.CF-1051-F-最短路+最小生成树2024-01-1726.CF-1184-E3-最小生成树+倍增+并查集2024-01-2427.CF-1304-E-倍增LCA+思维2024-01-2528.CF-1399-E2-优先队列2024-01-2229.CF-1515-F-思维2024-02-2430.CF-1831-E-卡特兰数+异或哈希+差分2024-01-2231.CF-1921-F-根号分治2024-01-24

收起

620-E 题目大意

给定一颗$n$个节点的树，根节点为$1$，每个节点都有一个颜色$c_i$。有$m$次操作，操作分两种：

• $1,x,c$：将以$x$为根的子树上的节点颜色都变为$c$。
• $2,x$：查询以$x$为根的子树上所有节点的颜色数量。

其中$1\le c,c_i\le 60$。

---

Solution

首先预处理出树的$dfs$序，把子树上的操作转换为区间操作，注意到颜色数量不超过$60$，可以用一个$64$位的整数来表示一个子树中所有的颜色。

转到$dfs$序上的操作后，操作一为把一段区间推平为$c$；操作二为询问区间中颜色数量，这里可以对区间中所有颜色取并，用位运算的或即可得到。至此，所有的操作都可以用懒标记线段树解决。

时间复杂度$O(nlogn)$，难点在是否了解前置知识$dfs$序和线段树。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=4e5+10;
struct node{
	int l,r,tag;
	ll cnt;
}tr[N<<2];

void pushdown(int u){
	if(tr[u].tag){
		tr[u<<1].cnt=tr[u<<1|1].cnt=1LL<<tr[u].tag;
		tr[u<<1].tag=tr[u<<1|1].tag=tr[u].tag;
		tr[u].tag=0;
	}
}

void build(int u,int l,int r,vector<int> &p){
	tr[u]={l,r,0,1LL<<p[l]};
	if(l==r) return;
	int m=(l+r)>>1;
	build(u<<1,l,m,p);
	build(u<<1|1,m+1,r,p);
	tr[u].cnt=tr[u<<1].cnt|tr[u<<1|1].cnt;
}

void modify(int u,int l,int r,int k){
	if(l<=tr[u].l&&tr[u].r<=r){
		tr[u].cnt=1LL<<k;
		tr[u].tag=k;
		return;
	}
	pushdown(u);
	int m=(tr[u].l+tr[u].r)>>1;
	if(l<=m) modify(u<<1,l,r,k);
	if(r>m) modify(u<<1|1,l,r,k);
	tr[u].cnt=tr[u<<1].cnt|tr[u<<1|1].cnt;
}

ll query(int u,int l,int r){
	if(l<=tr[u].l&&tr[u].r<=r){
		return tr[u].cnt;
	}
	pushdown(u);
	int m=(tr[u].l+tr[u].r)>>1;
	ll res=0;
	if(l<=m) res|=query(u<<1,l,r);
	if(r>m) res|=query(u<<1|1,l,r);
	return res;
}

void solve(){
	int n,m;
	cin>>n>>m;
	vector<int> col(n+1),p(n+1);
	for(int i=1;i<=n;i++) cin>>col[i];
	vector<vector<int>> e(n+1);
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	vector<int> L(n+1),R(n+1);
	int dfn=0;
	function<void(int,int)> dfs=[&](int x,int fa){
		L[x]=++dfn,p[dfn]=col[x];
		for(auto y:e[x]){
			if(y==fa) continue;
			dfs(y,x);
		}
		R[x]=dfn;
	};
	dfs(1,0);
	build(1,1,n,p);
	while(m--){	
		int op,x,c;
		cin>>op>>x;
		if(op&1){
			cin>>c;
			modify(1,L[x],R[x],c);
		}else{
			ll res=query(1,L[x],R[x]);
			cout<<__builtin_popcountll(res)<<'\n';
		}
	}
}

int main(){
	ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int T=1;
	//cin>>T;
	while(T--){
		solve();
	}
	return 0;
}