问题描述:给出一列数Ai,任选三个上升或下降的数为一组,问可以有这样的多少组。
数据范围:0<i<20000,ai<100000
分析:我觉得。。打死我也想不出这个用树状数组。。
问题转化成 求每个数左右两边比它小的数。因为扫一遍就已经n了。。再累计比它小的数,说明是遍扫描边更新。树状数组求和和改变的操作复杂度为O(log ai)。扫描到每个数时,我们可以计算sum(a[i]-1),再add(a[i],1).逆序同样处理一遍就可以了。
code:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
const int maxai=100000+10;
const int maxn=20000+10;
int leftles[maxn],rightles[maxn],a[maxn];
int leftless[maxai],rightless[maxai];
void init(){
memset(leftles,0,sizeof(leftles));
memset(a,0,sizeof(a));
memset(rightles,0,sizeof(rightles));
memset(leftless,0,sizeof(leftless));
memset(rightless,0,sizeof(rightless));
}
int lowbit(int x){ return x&-x; }
int sum(int x,int *s){
int ret=0;
while(x>0){
ret+=s[x];
x-=lowbit(x);
}
return ret;
}
void add(int x,int d,int *s){
while(x<=maxai){
s[x]+=d;
x+=lowbit(x);
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
int n;
scanf("%d",&n);
init();
for(int i=0;i<n;i++) scanf("%d",&a[i]);
for(int i=0;i<n;i++){
leftles[i]=sum(a[i]-1,leftless);
add(a[i],1,leftless);
}
for(int i=n-1;i>=0;i--){
rightles[i]=sum(a[i]-1,rightless);
add(a[i],1,rightless);
}
long long ans=0;
for(int i=0;i<n;i++){
ans=ans+(i-leftles[i])*rightles[i]+leftles[i]*(n-1-i-rightles[i]);
}
printf("%lld\n",ans);
}
return 0;
}