2020-11-8 一周总结 树状数组

树状数组

tr[i]的意义是 i-lowbit(i)+1到i长度为lowbit（i）的区间和

实现不难，难在做题时理清思路，知道该用树状数组

---

树状数组基本操作：

1.修改数列中某一个数 O(logn)
2.求前缀和 O(logn)
扩展操作：

1.修改连续区间的值
2.求某个点的值

问题基本类型：实质求某个区间和

基操：
lowbit（）2进制中排在末尾的1

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

add(int x,int c) 在a[x]上+c

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

sum(int x) 求区间和

int sum(int x)
{
	int ans=0;
	for(int i=x;i;i-=lowbit(i)) ans+=tr[i];
	return ans;
}

经典例题

eg 1：楼兰图腾
在完成了分配任务之后，西部314来到了楼兰古城的西部。

相传很久以前这片土地上(比楼兰古城还早)生活着两个部落，一个部落崇拜尖刀(‘V’)，一个部落崇拜铁锹(‘∧’)，他们分别用V和∧的形状来代表各自部落的图腾。

西部314在楼兰古城的下面发现了一幅巨大的壁画，壁画上被标记出了N个点，经测量发现这N个点的水平位置和竖直位置是两两不同的。

西部314认为这幅壁画所包含的信息与这N个点的相对位置有关，因此不妨设坐标分别为(1,y1),(2,y2),…,(n,yn),其中y1~yn是1到n的一个排列。

西部314打算研究这幅壁画中包含着多少个图腾。

如果三个点(i,yi),(j,yj),(k,yk)满足1≤i<j<k≤n且yi>yj,yj<yk，则称这三个点构成V图腾;
如果三个点(i,yi),(j,yj),(k,yk)满足1≤i<j<k≤n且yi<yj,yj>yk，则称这三个点构成∧图腾;

西部314想知道，这n个点中两个部落图腾的数目。

因此，你需要编写一个程序来求出v的个数和∧的个数。

输入格式
第一行一个数n。

第二行是n个数，分别代表y1，y2,…,yn。

输出格式
两个数，中间用空格隔开，依次为V的个数和∧的个数。

数据范围
对于所有数据，n≤200000,且输出答案不会超过int64。
y1∼yn 是 1 到 n 的一个排列。

输入样例：
5
1 5 3 2 4
输出样例：
3 4

难度： 简单
来源： 《算法竞赛进阶指南》

题意：

将n个点对应到二维平面，任选三个点，构成V或者∧的选法数

思路：

求每一个点左右两侧比他小的数的个数，或者比他大的数的个数，相乘相加。
将一个数组第i位为1表示i存在，为0表示不存在，树状数组就可以维护这个数组小于某个数的个数

#include<iostream>
#include<cstring>
using namespace std;
const int N = 2e5+10;
typedef long long ll;
int a[N],tri[N];
int low[N],more[N];
int n;
int lowerbit(int x)
{
    return x&-x;    
}

void add(int x,int c)
{
    for(int i=x;i<=n;i+=lowerbit(i)) tri[i]+=c;
}

int sum(int x)
{
    int ans=0;
    for(int i=x;i;i-=lowerbit(i)) ans+=tri[i];
    return ans;
}

int main()
{
    
    cin>>n;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    
    ll sum1=0,sum2=0;
    for(int i=1;i<=n;i++)
    {
        int t=a[i];
        more[i]=sum(n)-sum(t);//求左边比他大的数的个数
        low[i]=sum(t-1);//求左边比他小的数的个数
        add(t,1);//将这个数插入数组
    }
    memset(tri,0,sizeof tri);
    for(int i=n;i;i--)
    {
        int t=a[i];
        //求右边的
        sum1+=1ll*more[i]*(sum(n)-sum(t));
        sum2+=1ll*low[i]*sum(t-1);
        add(t,1);
    }
    cout<<sum1<<" "<<sum2<<endl;
    return 0;
}

扩展1

eg.一个简单的整数问题
题意：
n个操作，操作1把数列中第l~r个数都加d，操作2询问数列中第x个数的值

思路：
前缀和？O(1)查询，O(n)修改 复杂度O(nm)
差分？ O(1)修改，O(n)查询 复杂度O(nm)
树状数组+差分 O(logn)修改 O(logn)查询 复杂度O(mlogn)
树状数组维护差分数组的前缀和

#include<iostream>
using namespace std;
typedef long long ll;
const int N= 1e5+10;
int a[N],b[N];
ll tri[N];
int n,m;
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int d)
{
    for(int i=x;i<=n;i+=lowbit(i)) tri[i]+=d;
}

ll sum(int x)
{
    ll ans=0;
    for(int i=x;i;i-=lowbit(i)) ans+=tri[i];
    return ans;
}

int main()
{
    
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) b[i]=a[i]-a[i-1];
    //建树
    for(int i=1;i<=n;i++) add(i,b[i]);
    while(m--)
    {
        char op[2];
        int l,r,x;
        scanf("%s",op);
        if(op[0]=='C'){
          scanf("%d%d%d",&l,&r,&x);
          add(l,x);
          add(r+1,-x);
        }
        else{
            scanf("%d",&x);
            cout<<sum(x)<<endl;
        }
    }
    return 0;
}

扩展2

eg.一个简单的整数问题2
题意：
n个操作，操作1把数列中第l~r个数都加d，操作2询问数列中 l~r 个数的和
思路：
列出式子找到规律，先转换成求数列1~x的数的和，即差分数组的前缀和的前缀和
b1
b1+b2
b1+b2+b3
…
b1+b2+b3+…+bx
转换思路:补成x+1的矩阵
b1+b2+b3+…+bx
…
b1+b2+b3+…+bx
求（b1+b2+b3+…+bx）*(x+1) — bi * i的前x项和
维护两个树状数组，tr1维护差分数组的前缀和，tr2维护bi * i的前缀和

#include<iostream>

using namespace std;
typedef long long ll;
const int N = 1e5+10;
int n,m;
int a[N];
ll tr1[N],tr2[N];

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

void add(ll tr[],int x,ll d)
{
    for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=d;
}

ll sum(int x)
{
    ll ans1=0,ans2=0;
    for(int i=x;i;i-=lowbit(i)) ans1+=tr1[i],ans2+=tr2[i];
    
    return ans1*(1+x)-ans2;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    
    for(int i=1;i<=n;i++)
    {
        int b=a[i]-a[i-1];
        add(tr1,i,b);
        add(tr2,i,1ll*i*b);
    }
    
    while(m--)
    {
        int l,r,d;
        char op[2];
        scanf("%s%d%d",op,&l,&r);
        if(op[0]=='Q') printf("%lld\n",sum(r)-sum(l-1));
        else
        {
            scanf("%d",&d);
            add(tr1,l,d),add(tr1,r+1,-d);
            add(tr2,l,l*d),add(tr2,r+1,-1ll*(r+1)*d);
            //区间的修改实质是b[l]+d,b[r+1]-d中间b[i]的值未发生变化
            //对应b[i]*i的值也只在l和 r+1的位置发生变化
        }
    }
    return 0;
}