原题地址
题意:给一颗根为1的树,节点数为n,一开始每个节点上有一颗苹果,现有两个操作:
1.Q x,输出x的子树上的苹果个数。
2.C x,如果x点有苹果就拿走,没有就长一个出来。
m次操作。
思路:用dfs序把树上的节点转换成区间形式,然后用树状数组或者线段树维护和查询,模板题。
#include <iostream>
#include <stdio.h>
#include<cmath>
#include <cstdio>
#include<algorithm>
#include <cstring>
#include <string>
#define LL long long
#define INF INT64_MAX
#define MOD 1000000007
#define stree SegTree[root]
#define lson SegTree[root << 1]
#define rson SegTree[root << 1 | 1]
using namespace std;
typedef pair<int,int>pa;
const int N = 5e5+7;
char s[105];
int cnt, k, vis[N], d[N], head[N], l[N], r[N];
struct node
{
int to, next;
}edge[N];
void add(int u, int v)
{
edge[k].to = v;
edge[k].next = head[u];
head[u] = k++;
}
void dfs(int x)
{
l[x] = ++cnt;
for(int i = head[x];i!=-1;i = edge[i].next)
dfs(edge[i].to);
r[x] = cnt;
}
int lowbit(int x) {return x&(-x);}
void update(int x, int val)
{
for(int i = x;i <= cnt;i+=lowbit(i)) d[i] += val;
}
int sum(int x)
{
int res = 0;
for(int i = x;i; i-=lowbit(i)) res += d[i];
return res;
}
int main()
{
int t, n, m, x, k, u, v, q;
scanf("%d", &n);
memset(head, -1, sizeof(head));
for(int i = 1;i < n;i++)
{
scanf("%d %d", &u, &v);
add(u, v);
}
dfs(1);
for(int i = 1;i <= n;i++)
{
vis[i] = 1;
update(l[i], 1);
}
scanf("%d", &q);
while(q--)
{
scanf("%s%d", s, &x);
if(s[0]=='Q')
printf("%d\n", sum(r[x])-sum(l[x]-1));
else
{
if(vis[x]) update(l[x], -1);
else update(l[x], 1);
vis[x] ^= 1;
}
}
return 0;
}