poj3321 树状数组+DFS() 如果让我说:我只能说,实力决定一切

树状数组:对区间进行:①增②改③查
思路:该题就是求解区间的值(该区间苹果的个数),因此 我们需要处理树杈的问题,当前的树杈是由哪个树杈引出来的,我们需要找到起始树杈(当前树杈)和
结束的树杈,然后,在用树状数组去维护,进行查找,修改。用DFS()对当前树杈的起始点和终止点进行标号,类似二叉树的遍历(该题是多叉树)

#include<iostream>
#include<algorithm>
#include<string.h>
#include<cstdio>
using namespace std;
const int maxn=100010;
int num[maxn];
int st[maxn],ed[maxn];
struct node//利用广义表的结构
{
    int data;
    node *next;
} Node[maxn];
int N,M,t;
char c;
void DFS(int pox)//多叉树搜索。
{
    st[pox]=++t;
    node *tem=Node[pox].next;
    while(tem)
    {
        if(st[tem->data]==0)
        {
            DFS(tem->data);
        }
        tem=tem->next;
    }
    ed[pox]=t;
}
int lowbit(int x)
{
    return -x&x;
}
int get_sum(int pox)
{
    int sum=0;
    while(pox>0)
    {
        sum+=num[pox];
        pox-=lowbit(pox);
    }
    return sum;
}
void updata(int pox,int nu)
{
    while(pox<=N)
    {
        num[pox]+=nu;
        pox+=lowbit(pox);
    }
}
int main()
{
    int a,b;
    int tem;
    while(scanf("%d",&N)!=EOF)
    {
        memset(num,0,sizeof(num));
        t=0;
        for(int i=1; i<N; i++)
        {
            scanf("%d%d",&a,&b);
            node *p=new node();
            p->data=b;
            p->next=Node[a].next;
            Node[a].next=p;

            node *q=new node();
            q->data=a;
            q->next=Node[b].next;
            Node[b].next=q;
        }
        DFS(1);
        // printf("st[2]=%d\n",st[2]);
        for(int i=1; i<=N; i++)
        {
            updata(i,1);
        }
        for(int i=1; i<=N+1; i++)
            scanf("%d",&M);
        for(int i=0; i<M; i++)
        {
            cin>>c>>tem;
            if(c=='Q')
            {
                printf("%d\n",get_sum(ed[tem])-get_sum(st[tem]-1));
            }
            else
            {
                int sum=get_sum(st[tem])-get_sum(st[tem]-1);
                if(sum==1)
                    updata(st[tem],-1);
                else
                    updata(st[tem],1);
            }
        }
    }
    return 0;
}


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值