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 = 0;
    while(x > 0)
    {
        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 = 100010;
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,-1,sizeof(head));
    cnt = 0; num = 0;
    for(i=0; i<=n; i++)
    {
        c[i] = 0;
        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 = 0;
    while(x > 0)
    {
        ret += c[x];
        x -= lowbit(x);
    }
    return ret;
}
void dfs(int pos)
{
    s[pos] = ++num;
    for (int i = head[pos]; i != -1; i = g[i].next)
    {
        int v = g[i].v;
        dfs(v);
    }
    e[pos] = num;
}

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

//今天帮师兄做的笔试题，一个数组，求每个数前面比它大的个数
#include <iostream>
#include <cstring>
const int maxn = 2e5 + 10;
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 = 0;
    while(x > 0)
    {
        ret += c[x];
        x -= lowbit(x);
    }
    return ret;
}

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