区间信息的维护与查询1(树状数组)

动态连续和查询问题,可用树状数组(二叉索引树)快速完成。
每个节点只影响它的父节点。x的父节点为Lowbit(x)。
将这个树状数组存在数组c中。原数组a对后续Add(),Sum()没什么作用。

int Sum(int x)
{
    int ret = 0;
    while(x > 0) {
        ret += c[x];
        x -= lowbit(x);
    }
    return ret;
}
void Add(int x, int d)
{
    while(x <= n) {
        c[x] += d;
        x += lowbit(x);
    }
}

总时间复杂度为O(nlogn)

POJ 3321(dfs序+树状数组)

题意:有一颗苹果树,每个节点上有一个苹果,有两种操作:
C X,修改x点处的状态(从无到有,或从有到无);
Q X,查询x点及其子树分支共有多少苹果。

思路:每个节点都可以有多个子树,我们通过dfs来给每个节点编号。每个节点保存两个值,表示它的子树范围。然后把新的节点放进树状数组中更新查询。

#include<cstdio>
#include<cstring>
#include<vector>
#define lowbit(x) (x&-x)
using namespace std;
const int maxn = 100000 + 5;
vector<vector<int> > a(maxn);
int c[maxn],l[maxn],r[maxn],tot,s[maxn];
int n;

void add(int x, int k)
{
    for(int i = x; i <= n; i+=lowbit(i))
            c[i] += k;
}
int sum(int x)
{
    int ret = 0;
    for(int i = x; i > 0; i-=lowbit(i))
        ret += c[i];
    return ret;
}

void dfs(int x)
{
    l[x] = tot;
    for(int i = 0; i < a[x].size();i++) {
        tot++;
        dfs(a[x][i]);
    }
    r[x] = tot;
}

int main()
{
    int x, y, k, u, v, m;
    char op[3];
    while(~scanf("%d",&n)) {
        memset(c,0,sizeof(c));
        memset(s,0,sizeof(s));
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        for(int i = 0; i < n; i++) a[i].clear();

        for(int i = 1; i < n; i++) {
            scanf("%d%d",&u,&v);
            a[u].push_back(v);
        }
        tot = 1;
        dfs(1);//给每个节点编号,赋左右值
        for(int i = 1; i <= n; i++) {//把每个点加到树状数组中
            s[i] = 1;
            add(i,1);
        }
        scanf("%d",&m);
        while(m--) {
            scanf("%s%d",op,&x);
            if(op[0] == 'Q') {//查询
                printf("%d\n",sum(r[x])-sum(l[x]-1));
            }
            else {
                if(s[x]) add(l[x],-1);
                else add(l[x],1);
                s[x] = !s[x];
            }
        }
    }
    return 0;
}

【还有一点,vector< vector < int > > a(maxn);第一次知道后面用小括号。】

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值