poj 3321(带时间戳 + 区间统计)

题目描述：

给定一颗树，求某个节点的子树的val值之和

可以用树的前序遍历给每一个节点编号，从而可以确定一个节点的子树的范围，这样就可以进行直接在区间上进行统计了。

vector < int > Map[maxN]写成typedef vector <int> INT; vector <INT> Map(maxN);就不超时了。

线段树

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#define LL long long
using namespace std;
//线段树
//区间每点增值，求区间和
const int maxN = ;
struct node
{
    int lt, rt;
    int val;
}tree[*maxN];

//向上更新
void pushUp(int id)
{
    tree[id].val = tree[id<<].val + tree[id<<|].val;
}

//建立线段树
void build(int lt, int rt, int id)
{
    tree[id].lt = lt;
    tree[id].rt = rt;
   // tree[id].val = 1;//每段的初值，根据题目要求
    if (lt == rt)
    {
        tree[id].val=;
        return;
    }
    int mid = (lt+rt)>>;
    build(lt, mid, id<<);
    build(mid+, rt, id<<|);
    pushUp(id);
}

//增加区间内每个点固定的值
void add(int lt, int rt, int id)
{
    if (lt <= tree[id].lt && rt >= tree[id].rt)
    {
        if(tree[id].val==)
            tree[id].val=;
        else
            tree[id].val=;
        return;
    }
    int mid = (tree[id].lt+tree[id].rt)>>;
    if (lt <= mid)
        add(lt, rt, id<<);
    if (rt > mid)
        add(lt, rt, id<<|);
    pushUp(id);
}

//查询某段区间内的和
int query(int lt, int rt, int id)
{
    if (lt <= tree[id].lt && rt >= tree[id].rt)
        return tree[id].val;
    int mid = (tree[id].lt+tree[id].rt)>>;
    int ans = ;
    if (lt <= mid)
        ans += query(lt, rt, id<<);
    if (rt > mid)
        ans += query(lt, rt, id<<|);
    return ans;
}

struct NODE
{
    int cnt,l,r;
}a[maxN];
typedef vector<int > INT;
vector<INT > Map(maxN);
int vis[maxN];
int number;
void dfs(int fa)
{
    for(int i=;i<Map[fa].size();i++)
    {
        int son=Map[fa][i];
        if(vis[son]==)
        {
            a[son].cnt=++number;
            a[son].l= number;
            vis[son]=;
            dfs(son);
            a[son].r=number;
        }
    }
}
void init()
{
    for(int i=;i<maxN;i++)
        Map[i].clear();
}
int main()
{
   // freopen("test.txt","r",stdin);
    int n,m;
    while(~scanf("%d",&n))
    {
        init();
        for(int i=;i<=n-;i++)
        {
            int from,to;
            scanf("%d%d",&from,&to);
            Map[from].push_back(to);
            Map[to].push_back(from);
        }
        number=;
        memset(vis,,sizeof(vis));
        a[].cnt=;
        a[].l=;
        vis[]=;

        dfs();

        a[].r=number;

        build(a[].l,a[].r,);
        //for(int i=1;i<=9;i++)
        // printf("%d %d %d\n",a[i].cnt,a[i].l,a[i].r);
        int M;
        scanf("%d",&M);
        char str[];   int id;
        while(M--)
        {
            scanf("%s%d",str,&id);
            //printf("%s\n",str);
            if(str[]=='Q')
            {
                printf("%d\n",query(a[id].l,a[id].r,) );
            }
            if(str[]=='C')
            {
                add(a[id].cnt,a[id].cnt,);
            }
        }
    }
    return ;
}

树状数组

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#define LL long long
using namespace std;
const int maxN = ;
int n;
//树状数组
int C[maxN],A[maxN];
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret=;
    while(x>)
    {
        ret+=C[x]; x-=lowbit(x);
    }
    return ret;
}
void update(int x,int pls)
{
    int d;
    if(pls==)
    {
        if(A[x]==)
        d=;
        if(A[x]==)
        d=-;
    }
    else
        d=;
    A[x]+=d;
    while(x<=n)
    {
        C[x]+=d; x+=lowbit(x);
    }
}
struct NODE
{
    int cnt,l,r;
}a[maxN];
typedef vector<int>  INT;
vector < INT >   Map(maxN);
int vis[maxN];
int number;
void dfs(int fa)
{
    for(int i=;i<Map[fa].size();i++)
    {
        int son=Map[fa][i];
        if(vis[son]==)
        {
            a[son].cnt=++number;
            a[son].l= number;
            vis[son]=;
            dfs(son);
            a[son].r=number;
        }
    }
}
void init()
{
    for(int i=;i<maxN;i++)
        Map[i].clear();
}
int main()
{
    //freopen("test.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        init();
        for(int i=;i<=n-;i++)
        {
            int from,to;
            scanf("%d%d",&from,&to);
            Map[from].push_back(to);
            Map[to].push_back(from);
        }
        number=;
        memset(vis,,sizeof(vis));
        a[].cnt=;
        a[].l=;
        vis[]=;

        dfs();

        a[].r=number;
        memset(C,,sizeof(C));
        memset(A,,sizeof(A));
        for(int i=;i<=n;i++)
            update(i,);
        int M;
        scanf("%d",&M);
        char str[];   int id;
        while(M--)
        {
            scanf("%s%d",str,&id);
            if(str[]=='Q')
            {
                printf("%d\n",sum(a[id].r) -sum(a[id].l-) );
            }
            if(str[]=='C')
            {
                 update(a[id].cnt,);
            }
        }
    }
    return ;
}