POJ 3321 APPLE TREE（树状数组+DFS）

最新推荐文章于 2022-02-25 19:38:17 发布

acmj1991

最新推荐文章于 2022-02-25 19:38:17 发布

阅读量362

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/acmj1991/article/details/6690213

树状数组专栏收录该内容

3 篇文章

订阅专栏

解决了一个涉及苹果树节点状态更新与查询的问题，通过DFS遍历和子树编号映射实现子树内苹果数量的快速查询。

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

原文：http://blog.acmj1991.com/?p=669

题意：有一颗苹果树根为1，给出你每个叉点之间的关系。初始状态时每个叉点上有一个苹果。接下来有两种操作：C x 改变叉点x的状态（有改成无，无改成有）
Q x 查询x叉点的枝条中总共有多少个苹果（包括它本身）

思路：题目是插点问线，然后一直在想怎么生成这条线呢？而且它的子树的编号不一定连续，于是想到对子树编号进行映射，将子树编号编成连续。同时要记录子树的起始编号。查找子树的值就变成了查找子树区间内的值。
用dfs进行编号重新生成以及起点的记录
这题写了很久，但是1A了 300+ms 哈哈哈

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
 
#define maxN 100010
int head[maxN],st[maxN],pre[maxN],ke[maxN],vist[maxN],num;
struct point{
    int v,next;
}po[3*maxN];
 
void add(int u,int v)
{
    po[num].v=v;
    po[num].next=head[u],head[u]=num++;
}
void DFS(int s)
{
    st[s]=num;
    if(head[s]==-1){
        pre[s]=num++;
        return ;
    }
    for(int i=head[s];i!=-1;i=po[i].next)
        DFS(po[i].v);
    pre[s]=num++;
}
int low(int x){return x&(-x);}
int getsum(int x)
{
    int cnt=0;
    for(int i=x;i>0;i-=low(i))
        cnt+=ke[i];
    return cnt;
}
 
int main()
{
    char str[10];
    int n,m,u,v;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        ke[i]=low(i);
        vist[i]=1;
    }
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    num=1;
    DFS(1);
    scanf("%d",&m);
    while(m--){
        scanf("%s%d",str,&u);
        if(str[0]=='C'){
            int value=vist[u]?-1:1;
            vist[u]=!vist[u];
            for(int i=pre[u];i<=n;i+=low(i))
                ke[i]+=value;
        }else
            printf("%d\n",getsum(pre[u])-getsum(st[u]-1));
    }
}