暑假集训.10(树状数组模板)

一:Lowbit函数;lowbit这个函数的功能就是求某一个数的二进制表示中最低的一位1,举个例子,x = 6,它的二进制为110,那么lowbit(x)就返回2,因为最后一位1表示2

两种方法:

一:
int lowbit(x) 

{   

    return x - (x & (x - 1));

}
二:
int lowbit(x) 

{   

    return x & -x;

C数组用来存储叶子节点的权值:

  1. C[1]=A[1];

  2. C[2]=A[1]+A[2];

  3. C[3]=A[3];

  4. C[4]=A[1]+A[2]+A[3]+A[4];

  5. C[5]=A[5];

  6. C[6]=A[5]+A[6];

  7. C[7]=A[7];

  8. C[8]=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为i的二进制中从最低位到高位连续零的长度)例如i=8时,k=3;

#include<cstdio>
#include<iostream>
#include <cstring>
#include <math.h>
#include<algorithm>
using namespace std;
typedef long long ll;
ll a[1000000]= {0},c[1000000]= {0},n,i,m,dis[1000000],x,y;
//struct T
//{
//    ll x,y;
//} b[10000];
//bool cmp1(struct T a,struct T b)
//{
//    return a.x<b.x;
//}
//bool cmp2(struct T a,struct T b)
//{
//    return a.y<b.y;
//}
ll lowbit(ll i)
{
    return i&(-i);
}
ll Sum(ll i)//区间求和
{
    int ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
void Update(ll i,ll v)//更新
{
    while(i<=32100)
    {
        c[i]+=v;
        i+=lowbit(i);
    }
    return ;
}
int main()
{

    scanf("%lld",&n);
   memset(dis,0,sizeof(dis));

    memset(c,0,sizeof(c));
    for(i=1; i<=n; i++)
    {
        scanf("%lld%lld",&x,&y);
        x++;
        dis[Sum(x)]++;
        Update(x,1);

    }
    for (i = 0; i <= n -1; i++)
        printf("%lld\n",dis[i]);

//    for(i=1;i<=n;i++)// 离散化
//    {
//        scanf("%lld",&a[i]);
//        b[i].x=a[i];
//        b[i].y=i;
//        Update(i,a[i]);
//    }
//
//    sort(b+1,b+n+1,cmp1);
//    for(i=1;i<=n;i++)
//    {
//        b[i].x=i;
//    }
//    sort(b+1,b+n+1,cmp2);
//    for(i=1;i<=n;i++)
//    {
//        dis[i]=b[i].x;
//        Update(i,dis[i]);
//    }
//    ll ans=0;
//    for(i=1;i<=n;i++)
//    {
//        Update(dis[i],1);
//        ans+=i-Sum(dis[i]);
//    }
//    for(i=1;i<=n;i++)
//        printf("%lld\n",dis[i]);
//    for(i=1;i<=n;i++)
//        printf("%lld ",c[i]);
//    printf("%lld\n",ans);

    return 0;
}

利用树状数组可以求逆序对个数,代码如下:


#include<iostream>

#include <algorithm>

#include <cstdlib>

#include<cstdio>

#include<cstring>

#include <cmath>

using namespace std;

const int M=200000+10;

const int MAX=0x3f3f3f3f;

typedef long long ll;

ll a[101010],c[101010],t[101010],n;

int lowbit(ll x)

{

    return x&(-x);

}

int Getsum(ll i)//求前几项和

{

    ll ans=0;

    while(i>0)

    {

        ans+=c[i];

        i-=lowbit(i);

    }

    return ans;

}

void update(ll i,ll v)//更新

{

    while(i<=n)

    {

        c[i]+=v;

        i+=lowbit(i);

    }

}

int main()

{

    ll m,i,j;

    scanf("%lld",&n);

    for(i=1; i<=n; i++)

     {

        scanf("%lld",&a[i]);

        t[i]=a[i];

        //update(i,a[i]);//把c[i]求出来

     }

    sort(t+1,t+n+1);

    m=unique(t+1,t+n+1)-t-1;

    for(i=1; i<=n; i++)

        a[i]=lower_bound(t+1,t+m+1,a[i])-t;

//    for(int i=1;i<=m;i++)

//        printf("%d\n",a[i]);

    ll ans=0;

    for(i=1;i<=n;i++)

    {

         update(a[i],1);

         ans+=i-Getsum(a[i]);

    }

    printf("%lld\n",ans);

    return 0;

}

区间a,b都加2 求区间x,y的和(区间修改 区间查询模板):


#include<iostream>

#include <algorithm>

#include <cstdlib>

#include<cstdio>

#include<cstring>

#include <cmath>

using namespace std;

const int M=200000+10;

const int MAX=0x3f3f3f3f;

typedef long long ll;

ll a[101010],c[101010],t[101010],n=9;

ll aa[10000]={0},bb[10000]={0};

ll lowbit(ll x)

{

    return x&(-x);

}

ll Getsum(ll i,ll q[])//求前几项和

{

    ll ans=0;

    while(i>0)

    {

        ans+=q[i];

        i-=lowbit(i);

    }

    return ans;

}

void update(ll i,ll v,ll q[])//更新

{

    while(i<=n)

    {

        q[i]+=v;

        i+=lowbit(i);

    }

}

int main()

{

    //a,b区间都加2 求x,y区间的和 视最初都为0 所以只是求增加的和

    ll a,b,x,y,i,j,ans=0;

    memset(aa,0,sizeof(aa));

    memset(bb,0,sizeof(bb));

    scanf("%lld%lld%lld%lld",&a,&b,&x,&y);

    update(a,2,aa);

    update(b+1,-2,aa);

    update(a,a*2,bb);

    update(b+1,-(b+1)*2,bb);

    ans+=(y+1)*Getsum(y,aa)-Getsum(y,bb);

    ans-=x*Getsum((x-1),aa)

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值