POJ 3321 【树状数组+dfs序】

POJ 3321 【树状数组+dfs序】

题目链接

题意：树上有N个节点，卡卡给他们编号1到N，根的编号永远是1.每个节点上最多结一个苹果。卡卡想要了解某一个子树上一共结了多少苹果。
“C x” 表示某个节点上的苹果发生了变化，如果原来没有苹果，则现在长出了一个苹果；如果原来有苹果，则是卡卡把它吃了。
“Q x” 表示查询x节点上的子树上的苹果有多少。包含节点x.

思路：第一个操作直接O(1)修改？，但是查询的时候怎么办，还要更新各节点？很像树状数组操作啊，第一个操作直接add(id,1),只是查询的时候查询的是区间和，现在变成了子树，现在要解决的问题就是把树上问题转化成区间序列，当然就是dfs序：一遍dfs可以记录每个节点的位置及其进出的时间，位置就是id,那么子树不就是位置进出之间的id吗？这不就是查询区间和问题吗？

注意点：vector<vector >v(maxn);——>正确写法
vectorv[maxn];会超时

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cstdio>
using namespace std;
const int maxn=1e5+5;
vector<vector<int> >v(maxn);
int num[maxn],st[maxn],ed[maxn],cnt=0;
int c[maxn];
int n,x,y;
char op[2];
int lowbit(int u)
{
    return u&(-u);
}
void dfs(int now,int fa)
{
    st[now]=++cnt;
    num[cnt]=now;
    for(int i=0; i<v[now].size(); i++)
    {
        if(v[now][i]==fa)
            continue;
        dfs(v[now][i],now);
    }
    ed[now]=cnt;
}
void add(int id,int val)
{
    while(id<=n)
    {
        c[id]+=val;
        id+=lowbit(id);
    }
}
int query(int id)
{
    int ans=0;
    while(id)
    {
        ans+=c[id];
        id-=lowbit(id);
    }
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1; i<n; i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1,-1);
    for(int i=1; i<=n; i++)
    {
        add(st[i],1);
    }
    int q;
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s",op);
        if(op[0]=='Q')
        {
            scanf("%d",&x);
            printf("%d\n",query(ed[x])-query(st[x]-1));
        }
        else
        {
            int flag=-1;
            scanf("%d",&x);
            if(query(st[x])-query(st[x]-1)==0)
            {
                flag=1;
            }
            add(st[x],flag);
        }
    }
}

---