dfs+线段树 zhrt的数据结构课

Posted 2022-06-30 echozqn

zhrt的数据结构课

这个题目我觉得是一个有一点点思维的dfs+线段树

虽然说看起来可以用树链剖分写，但是这个题目时间卡了树剖

因为之前用树剖一直在写这个，所以一直想的是区间更新，想dfs+线段树，有点点没想明白

后来才知道可以把这个区间更新转化成单点更新，就是查一个结点的子树，如果子树有可以到根节点的，那么这个结点肯定也可以到根节点。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <stack>
#include <map>
#include <string>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 4e5 + 10;
int sum[maxn * 4];

void push_up(int id)

	sum[id] = sum[id << 1 | 1] + sum[id << 1];


void build(int id,int l,int r)

	if(l==r)
	
		sum[id] = 0;
		return;
	
	int mid = (l + r) >> 1;
	build(id << 1, l, mid);
	build(id << 1 | 1, mid + 1, r);
	push_up(id);


void update(int id,int l,int r,int pos,int val)

	if(l==r)
	
		sum[id] += val;
		return;
	
	int mid = (l + r) >> 1;
	if (pos <= mid) update(id << 1, l, mid, pos, val);
	else update(id << 1 | 1, mid + 1, r, pos, val);
	push_up(id);


int query(int id,int l,int r,int x,int y)

	// printf("id=%d l=%d r=%d x=%d y=%d\n", id, l, r, x, y);
	if (x <= l && y >= r) return sum[id];
	int mid = (l + r) >> 1;
	int ans = 0;
	if (x <= mid) ans += query(id << 1, l, mid, x, y);
	if (y > mid) ans += query(id << 1 | 1, mid + 1, r, x, y);
	return ans;

int el[maxn], er[maxn], tot = 0, head[maxn], cnt;
struct node

	int v, nxt;
	node(int v=0,int nxt=0):v(v),nxt(nxt)
ex[maxn];

void init()

	memset(head, -1, sizeof(head));
	tot = 0, cnt = 0;


void add(int u,int v)

	ex[cnt] = node(v, head[u]);
	head[u] = cnt++;
	ex[cnt] = node(u, head[v]);
	head[v] = cnt++;
	// printf("u=%d v=%d\n", u, v);


void dfs(int u,int pre)

	el[u] = ++tot;
	for(int i=head[u];i!=-1;i=ex[i].nxt)
	
		int v = ex[i].v;
		if (v == pre) continue;
		dfs(v, u);
	
	er[u] = tot;
	// printf("el[%d]=%d er[%d]=%d\n", u, el[u], u, er[u]);


int main()

	int t;
	scanf("%d", &t);
	while(t--)
	
		init();
		int n, m;
		scanf("%d%d", &n, &m);
		build(1, 1, n);
		for(int i=1;i<n;i++)
		
			int u, v;
			scanf("%d%d", &u, &v);
			add(u, v);
		
		dfs(1, -1);
		while(m--)
		
			int opt, x;
			scanf("%d%d", &opt, &x);
			if (opt == 0) update(1, 1, n, el[x], 1);
			if (opt == 1) update(1, 1, n, el[x], -1);
			if (opt == 2)
			
				int ans = query(1, 1, n, el[x], er[x]);
				if (ans) printf("Yes\n");
				else printf("No\n");