Apple Tree

最新推荐文章于 2022-03-19 17:21:38 发布

皮得一1

最新推荐文章于 2022-03-19 17:21:38 发布

阅读量457

点赞数

CC 4.0 BY-SA版权

分类专栏： oj题目

本文链接：https://blog.youkuaiyun.com/u014552756/article/details/38317149

oj题目专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种结合树状数组和深度优先搜索（DFS）的算法实现，用于解决图论问题中的节点查询与更新操作。通过对树结构的遍历，并利用树状数组高效地维护节点状态，该方法能够在对数时间内完成区间求和等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <stdio.h>
using namespace std;
#define MAXN 100010
int N;
int cnt=0;
int c[MAXN];
int start[MAXN];
int end[MAXN];
struct Node
{
    int num;
    Node* next;  //孩子节点
    Node()
    {next = NULL;}
}tree[MAXN];    //临界表

int lowbit(int x)
{
    return x & (-x);
}

void add(int d,int x)
{
    while(d<=N){
        c[d] += x;
        d += lowbit(d);
    }
}

int sum(int d)
{
    int ans =0;
    while(d>=1){
        ans += c[d];
        d -= lowbit(d);
    }
    return ans;
}

void dfs(int v)    //以r为根节点进行dfs遍历，返整个遍历之后的时间
{
    start[v] = ++cnt;
    Node* p = tree[v].next;
    while(p){
        if(start[p->num]==0)
            dfs(p->num);
        p = p->next;
    }
    end[v] = cnt;
}

void addedge(int a,int b)    //在苹果树上加分支
{
    Node* p = new Node;
    p->num = b;
    p->next = tree[a].next;
    tree[a].next = p;
}

int main()
{
    int i,q;
    scanf("%d",&N);
    for(i=1;i<N;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        addedge(a,b);
        addedge(b,a);
    }
    dfs(1);

    for(i=1;i<=N;i++)    //初始化c[]
        add(i,1);

    scanf("%d",&q);
    while(q--){    //q次操作
        char cmd[10];
        int d;
        scanf("%s%d",cmd,&d);
        if(cmd[0]=='C'){
            if(sum(start[d])-sum(start[d]-1)==1)
                add(start[d],-1);
            else
                add(start[d],1);
        }
        else if(cmd[0]=='Q'){
            printf("%d\n",sum(end[d])-sum(start[d]-1));
        }
    }
    return 0;
}