题目
这题乍一看感觉就是一个逆序对的问题,和树状数组好像没有什么关系,毕竟树状数组的优势就只有动态的单点修改和区间查询,但是仔细推理还是可以发现关系的(反正我是发现不了。。。),因为这个题是求逆序对,我们其实只需要找到每个小朋友在他前面比他高的和在他后面比他矮的再计算即可。
所以需要两步操作:
(1)枚举每个小朋友
(2)分别找出在他前面比他高的和在他后面比他矮的小朋友。
之后就是数学公式了。
第一步没什么好想的,主要是第二步,如何找出在他前面比他高的呢,这就可以用到树状数组了,(这里想了好久,我真笨)我们可以用树状数组存每个高度出现的次数。因为我们是按小朋友的顺序枚举的,所以在枚举第x位小朋友之前已经往树状数组中放了x-1个数,分别是前x-1个小朋友的高度,我们只需在枚举第x位小朋友之前查询树状数组中属于h[x] (这是指第x为小朋友的高度) ----N-1这个区间共有多少个数即可。找到之后再将第x为小朋友的高度放入树状数组中。因为每枚举一位小朋友之间都需要查找一遍区间,并且需要对区间进行更新操作,所以用树状数组来做在合适不过了。
求在第x位小朋友后面且比这名小朋友低的小朋友的个数同理,只不过需要逆序枚举每位小朋友。
代码如下:
#include<iostream>
#include<cstring>
using namespace std;
const int N=1e6+10;
long long a[N],b[N],tr[N];
int lowbit(int x){
return x&-x;
}
void add(int x,int c){
for(int i=x;i<N;i+=lowbit(i))tr[i]+=c;
}
int find(int x){
int ans=0;
for(int i=x;i;i-=lowbit(i))ans+=tr[i];
return ans;
}
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++)scanf("%lld",&a[i]);
for(int i=0;i<n;i++){
b[i]=find(N-1)-find(a[i]);
add(a[i],1);
}
memset(tr,0,sizeof tr);
for(int i=n-1;~i;i--){
b[i]+=find(a[i]-1);
add(a[i],1);
}
long long ans=0;
for(int i=0;i<n;i++)ans+=(b[i]*(b[i]+1)>>1);
cout<<ans;
return 0;
}