poj 3321 Apple Tree(一维树状数组)

题目：http://poj.org/problem?id=3321

题意：

苹果树上n个分叉，Q是询问，C是改变状态。。。。

开始的处理比较难，参考了一下大神的思路，构图成邻接表 并 用DFS编号

白书上一维树状数组模板：

 int lowbit(int x)
 {
     return x&(-x);
 }
 void add(int x,int d)   //c[]的下标要从 1开始。
 {
     while(x <= n)
     {
         c[x] += d;
         x +=lowbit(x);
     }
 }
 int sum(int x)  //前x项的和。
 {
     int ret = ;
     while(x > )
     {
         ret += c[x];
         x -= lowbit(x);
     }
     return ret;
 }

AC代码：

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <cstdlib>
 #include <algorithm>
 using namespace std;
 const int maxn = ;
 int s[maxn],e[maxn],c[maxn],num;
 int head[maxn],n,cnt;
 bool vis[maxn];
 struct node
 {
     int v,next;
 }g[maxn];

 void init()
 {
     int i;
     memset(head,-,sizeof(head));
     cnt = ; num = ;
     for(i=; i<=n; i++)
     {
         c[i] = ;
         vis[i] = false;
     }
 }
 void add_e(int u,int v)
 {
     g[cnt].v = v;
     g[cnt].next = head[u];
     head[u] = cnt++;
 }
 int lowbit(int x)
 {
     return x&(-x);
 }

 void add(int x,int d)
 {
     while(x <= n)
     {
         c[x] += d;
         x +=lowbit(x);
     }
 }
 int sum(int x)
 {
     int ret = ;
     while(x > )
     {
         ret += c[x];
         x -= lowbit(x);
     }
     return ret;
 }
 void dfs(int pos)
 {
     s[pos] = ++num;
     for (int i = head[pos]; i != -; i = g[i].next)
     {
         int v = g[i].v;
         dfs(v);
     }
     e[pos] = num;
 }

 int main()
 {
     int x,y,i,q;
     char op[];
     scanf("%d",&n);
     init();
     for(i = ; i < n-; i++)
     {
         cin>>x>>y;
         add_e(x,y); //邻接表构图
     }
     dfs(); //编号
     for (i = ; i <= n; i++)
     add(i,);
     cin>>q;
     while (q--)
     {
         cin>>op>>x;
         if (op[] == 'Q')
             printf("%d\n",sum(e[x]) - sum(s[x] - ));  //输出连续的和
         else       //改变状态
         {
             if (!vis[x])
             {
                 add(s[x],-);
                 vis[x] = true;
             }
             else
             {
                 add(s[x],);
                 vis[x] = false;
             }
         }
     }
     return ;
 }

 //今天帮师兄做的笔试题，一个数组，求每个数前面比它大的个数
 #include <iostream>
 #include <cstring>
 const int maxn = 2e5 + ;
 using namespace std;

 int c[maxn], n = maxn;

 int lowbit(int x)
 {
     return x&(-x);
 }
 void add(int x,int d)
 {
     while(x <= n)    //这里的n指c数组总数
     {
         c[x] += d;
         x +=lowbit(x);
     }
 }
 int sum(int x)  //求c[1]到c[x]的和
 {
     int ret = ;
     while(x > )
     {
         ret += c[x];
         x -= lowbit(x);
     }
     return ret;
 }

 int main()
 {
     int i, t, n1;
     while(cin>>n1)
     {
         memset(c, , sizeof(c));
         for(i = ; i <= n1; i++) //c数组从1开始
         {
             cin>>t;
             add(t, );
             if(i != n1)
             cout<<sum(maxn-)-sum(t)<<" ";
             else
             cout<<sum(maxn-)-sum(t)<<endl;
         }
     }
     return ;
 }