树状数组（例洛谷p1673）

树状数组

树状数组算是一个非常奇葩的高级数据结构，(ge人理解）,相对于线段树理解起来更加麻烦，但是代码实现却是比较简单

这里以洛谷p1637作为例题

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;
const int N=1e5+10;
typedef long long LL; 
int n;
int ma=0;
LL a[N];
LL tr[N*4];
LL  le[N],ri[N];//分别记录比第i个数小的但是出现在i左边，和比第i个数大，但是出现在第i个数右边
int lowbit(int x){//返回x二进制表示中第一个1的值如5=0101，则返回1
    return x&-x;
}
void add(LL x,LL c){//单点修改，在x值的数量加c,(表示在第x个书上加c)
    for(int i=x;i<=ma;i+=lowbit(i))tr[i]+=c;//所以我们的区间的最终点在ma,最大的a[i],而不是n
}
LL sum(int x){//求1到x出现的数量的和
    LL res=0;
    for(int i=x;i;i-=lowbit(i))res+=tr[i];
    return res;
}
int main(){
    ios::sync_with_stdio(false);
   
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]>ma)ma=a[i];
    }
    for(int i=1;i<=n;i++){
         le[i]=sum(a[i]-1);//比第i个数小的但是出现在i左边
        add(a[i],1);
        
    }
    memset(tr,0,sizeof tr);//初始化，重新来一遍
 
    for(int i=n;i>0;i--){
        ri[i]=sum(ma)-sum(a[i]);//比第i个数大，但是出现在第i个数右边
        add(a[i],1);
       
        
    }
    LL ans=0;
    for(int i=1;i<=n;i++ ){
        ans+=(le[i]*ri[i]);
    }
    
    cout<<ans;
    return 0;
}

应为我们要找一个三个数，而且得是连续徐上升的三个子序列，所以我们只需要记录比第i个数小的但是出现在i左边，和比第i个数大，但是出现在第i个数右边，然后ans+=(le[i]*ri[i])，就是最终的答案

这题还有一个线段树的做法，如果有需求·评论区·留言