树状数组模板加理解

树状数组2的关键是lowbit()函数,lowbit(x)=x&-x,功能是找到x的二进制的最后一“1”(不得不感叹这个精妙之处),其原理是利用负数的补码,补码是源码取反加一,例如x=6=000110(2),-x=x补=111010(2),那么lowbit(x)=2;
大致上树状数组是运用三个函数来实现lowbit(),add(),sum(),

lowbit(x)引出一个tree[ ]数组,所有的计算都是围绕tree[ ]进行,

#define lowbit(x) ((x)& -(x))

void add(int x,int d){//单点修改
    while(x<=n){
        tree[x]+=d;
        x+=lowbit(x);
    }
}

int sum(int x){//求区间1-x的和
    int sum=0;
    while(x>0){
        sum+=tree[x];
        x-=lowbit(x);
    }
    return sum;
}

poj 2182为例
(1)在n个位置上,每个位置上都有一头牛,及a1=a2=…=an=1,不过并不需要直接定义和使用数组a[ ]
(2)tree[ ]数组的初始化,这个题目比较特殊,不需要add初始化,因为lowbit()就是tree[i];
(3)程序所要做的,就是对每一个x[i]+1,用find_cow()找出对应的x,就是第x头牛之后,ax=0,方法就是add()更新数组tree[ ],及执行add(x,-1),
下面的程序完全套模板

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<string.h>
#define ll long long
#include<queue>
#include<map>
#include<set>
#define inf 0x3f3f3f3f
#define lowbit(x) ((x)& -(x))
using namespace std;
int n,tree[100005],x[100005];
int ans[100005];
void add(int x,int d){
    while(x<=n){
        tree[x]+=d;
        x+=lowbit(x);
    }
}
int sum(int x){
    int sum=0;
    while(x>0){
        sum+=tree[x];
        x-=lowbit(x);
    }
    return sum;
}
int find_cow(int x){
    int l=1,r=n;
    while(l<r){
        int mid=(l+r)>>1;
        if(sum(mid)<x)l=mid+1;
        else r=mid;
    }
    return l;

}
int main(){
    cin>>n;
    x[1]=0;
    for(int i=2;i<=n;i++){
        scanf("%d",&x[i]);
    }
    for(int i=1;i<=n;i++){
        tree[i]=lowbit(i);
    }
    for(int i=n;i>0;i--){
        int r=find_cow(x[i]+1);
        add(r,-1);
        ans[i]=r;
    }
    for(int i=1;i<=n;i++){
        printf("%d\n",ans[i]);
    }
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值