树状数组模板

网上关于树状数组的介绍很多,这里就不详细介绍,主要是总结模板
首先lowbit部分

int lowbit(int x) {return x&(-x);}

树状数组的标准用法是单点修改求前缀和
普通的用法如下(复杂度均为logn)

单点修改` 区间查询(求前缀和)初始化的话执行n次add操作

void add(int x,int k){
    while(x<=n){
        c[x]+=k;
        x+=lowbit(x);
    }
}

int query(int x){
    int sum=0;
    while(x>0){
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}

然后是区间更新,单点查询 这里使用差分的思想,维护差分数组c[i]
因为对一段区间都加上k ,他们他们之间的差值是不会变的,只有c[l] 和c[r+1]发生了变化,而数组c[i]的前缀和正好是a[i]
在这里插入图片描述
代码类似,只是修改的时候应该是add(l,k)和add(r+1,-k)

最后就是区间修改和区间查询 和上面的思路类似
在这里插入图片描述

void add(int x,int k){
    int p=x;
    while(x<=n){
        sum1[x]+=k
        sum2[x]+=k*(p-1); //注意是p
        x+=lowbit(x);
    }
}
void range_add(int l,int r,int k){
    add(l,k);
    add(r+1,-k);
}
int query(int x){
    int p = x;
    int sum=0;
    while(x>0){
        sum+=p*sum1[x]-sum2[x];
        x-=lowbit(x);
    }
    return sum;
}
int range_query(int l,int r){
    return query(r)-query(l-1); // 包括当前下标
}

poj 2299 树状数组求逆序对模板

#include <iostream>
#include <string.h>
#include <algorithm>

using namespace std;
const int MAX=5e5+5;
struct node{
    int v;
    int i;
}a[MAX];
int c[MAX];
int n;
bool cmp(const struct node a,const struct node b){
    if(a.v==b.v) return a.i<b.i;
    return a.v<b.v;
}
int lowbit(int x){return x&(-x);}
void add(int x,int k){
    while(x<=n){
        c[x]+=k;
        x+=lowbit(x);
    }
}
int query(int x){
    int sum=0;
    while(x>0){
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main (){
    //int n;
    while(cin>>n&&n){
        memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++){
            cin>>a[i].v;
            a[i].i=i;
        }
        sort(a+1,a+n+1,cmp);
        long long ans=0;
        for(int i=1;i<=n;i++){
            add(a[i].i,1);
            ans+=i-query(a[i].i);
        }
        cout<<ans<<endl;
    }



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值