POJ 3321 APPLE TREE(树状数组+DFS)

解决了一个涉及苹果树节点状态更新与查询的问题,通过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));
    }
}


 


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值