POJ3321 Apple Tree dfs序+树状数组

原创于 2021-08-08 21:29:03 发布 · 184 阅读

1 ·

CC 4.0 BY-SA版权

dfs 专栏收录该内容

4 篇文章

订阅专栏

这是一篇关于树状数组或线段树在解决树上节点苹果数量查询与修改问题的程序设计文章。通过深度优先搜索将树转换为区间形式，并使用树状数组进行区间维护，实现对节点苹果数的增减操作和查询。

原题地址
题意：给一颗根为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;
}