码迷,mamicode.com
首页 > 编程语言 > 详细

树状数组

时间:2019-08-18 13:28:34      阅读:87      评论:0      收藏:0      [点我收藏+]

标签:color   clu   lld   ref   引入   传统   targe   one   style   

树状数组:二进制的应用

与线段树的区别:树状数组的问题都可以用线段树解决,树状数组系数少,效率高

修改、查询复杂度 :O(log N)

技术图片

单点更新、区间查询:

C[1]=C[0001]=A[1]

C[2]=C[0010]=A[1]+A[2]

C[3]=C[0011]=A[3]

C[4]=C[0100]=A[1]+A[2]+A[3]+A[4]

 

C[5]=C[0101]=A[5]

C[6]=C[0110]=A[5]+A[6]

C[7]=C[0111]=A[7]

C[8]=C[1000]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6]+A[7]+A[8]

 

C[i]=A[i-2^k+1]+A[i-2^k+2]+...+A[i]

k:二进制中最低位到高位连续零的长度

 

Sum(i)=C[i]+C[i-2^k1]+C[(i-2^k1)-2^k2)+...

A[i] 包含于 C[i + 2k]、C[(i + 2k) + 2k]...

例如求前7项:C[7]+C[6]+C[4]

Sum[7]=sum[111]=C(100)+C(110)+C(111)

 

lowbit:取2^k

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

更新函数

void updata(int x,int y) {
    for(int i=x;i<=n;i+=lowbit(i))
    c[i]+=y;
}

求和函数

int getsum(int x) {
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
} 

例题http://acm.hdu.edu.cn/showproblem.php?pid=1166

技术图片
#include <bits/stdc++.h>
using namespace std;
int n,m;
int a[50005],c[50005];

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

void updata(int x,int y) {
    for(int i=x;i<=n;i+=lowbit(i))
    c[i]+=y;
}

int getsum(int x) {
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
} 

int main( ) {
    int t;
    cin>>t;
    int tot=1;
    while(t--) {
        cout << "Case " << tot++ << ":" << endl;
        memset(a, 0, sizeof a);
        memset(c, 0, sizeof c);
        
        cin>>n;
        
        for(int i = 1; i <= n; i++){
            cin>>a[i];
            updata(i,a[i]); 
        }

        string s;
        int x,y;
        
        while(cin>>s && s[0] != E) {
            
            cin>>x>>y;
            
            if(s[0] == Q) {
                int sum = getsum(y) - getsum(x-1);
                cout << sum << endl;
            }
            
            else if(s[0] == A) updata(x,y);
            
            else if(s[0] == S) updata(x,-y);
            
        }
        
    }
    return 0;
}
View Code

 

区间更新、单点查询:

a-b区间内所有值全部加上k或者减去k

用传统树状数组复杂度不允许,不能再用数据的值建树

引入差分,利用差分建树

C[i]=a[i]-a[i-1] a为原数组

 

当某个区间[x,y]的值改变,区间内的差值不变

只有C[x]和C[y+1]的值改变了

C[]数组建立树状数组

A[i]=Σ(i,j=1)C[i]  前面i项的差值和

 

例如:

A[0]=0;

 

A[]=1 2 3 5 6 9

C[]=1 1 1 2 1 3

把区间[3,5]加上2

A[]=1 2 5 7 8 9

C[]=1 1 3 2 1 1

 

C[3]的差值与C[5+1]的差值发生了改变

C[]数组建立树状数组

例题https://www.luogu.org/problem/P3368

技术图片
#include <iostream>
using namespace std;
typedef long long ll;
int a[500007],c[500007];
int n,m;
int lowbit(int x) {
    return x&(-x);
}
void update(int x,int y) {
    for(int i=x;i<=n;i+=lowbit(i)) c[i]+=y;
}

int getsum(int x) {
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}
int main( ) {
    scanf("%d %d",&n,&m);
    a[0]=0;
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        update(i,a[i]-a[i-1]);
    }
    
    for(int i=1;i<=m;i++) {
        int flag,x,y;
        int k;
        scanf("%d",&flag);
        if(flag==1) {
            scanf("%d %d %d",&x,&y,&k);
            update(x,k);
            update(y+1,-k);
        }
        else {
            scanf("%d",&x);
            printf("%d\n",getsum(x));
        }
    }
    
    return 0;
}
View Code

区间更新、区间查询

还是利用差分

Σ(n,i=1)A[i]=Σ(n,i=1)Σ(i,j=1)D[j]

即:

A[1]+A[2]+...+A[n]

=(D[1])+(D[1]+D[2])+...+(D[1]+D[2]+..+D[n])

=n*D[1]+(n-1)*D[2]+...+D[n]

=n*(D[1]+D[2]+...+D[n])-(0*D[1]+1*D[2]+...+(n-1)*D[n])

 

所以: Σ(n,i=1)A[i]=n*Σ(n,i=1)D[i] - Σ(n,i=1)(D[i]*(i-1));

维护两个树状数组

A[i]=D[i]

B[i]=D[i]*(i-1);

例题https://vjudge.net/problem/POJ-3468

技术图片
#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
ll a[500007],c[500007];
ll A[500007],B[500007];
int n,m;
int lowbit(int x) {
    return x&(-x);
}
void update(ll x,ll y) {
    ll k=x;
    for(int i=x;i<=n;i+=lowbit(i)) {
        A[i]+=y;
        B[i]+=y*(k-1);
    }
}

ll getsum(ll x) {
    ll ans=0;
    ll k=x; 
    for(int i=x;i>0;i-=lowbit(i)) ans+=k*A[i]-B[i];
    return ans;
}
int main( ) {
    scanf("%lld %lld",&n,&m);
    a[0]=0;
    for(int i=1;i<=n;i++) {
        scanf("%lld",&a[i]);
        update(i,a[i]-a[i-1]);
    }
    
    for(int i=1;i<=m;i++) {
        char s;
        cin>>s;
        ll x,y,k;
        if(s==Q) {
            scanf("%lld %lld",&x,&y);
            cout<<getsum(y)-getsum(x-1)<<endl;
        }
        else {
            scanf("%lld %lld %lld",&x,&y,&k);
            update(x,k);
            update(y+1,-k);
        }
    }
    
    return 0;
}
View Code

树状数组还可以求逆序对、区间最大值,这些以后再更新了

树状数组

标签:color   clu   lld   ref   引入   传统   targe   one   style   

原文地址:https://www.cnblogs.com/wronin/p/11372037.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!