输入样例:
5
1 5 3 2 4
输出样例:
3 4
解析:
以某个点 i 为最低点的 V 的数量,为 i 左侧和右侧比 a[ i ] 大的数量 a,b 的乘积。
但是,直接求这两个数的复杂度为O(n),则整个复杂度为O( n^2 ),数据量2e5超时。
所以需要将查询a,b两个数的复杂度讲到 logn 以下。
树状数组的下标储存 a[ i ],值储存 a[ i ] 的个数,所以,先计算 a[ i ] 左侧比 a[ i ] 大的数量,再计算右侧的数量,乘积加和即可。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int n,great[N],low[N],c[N],a[N];
int lowbit(int x){return x&-x;}
void add(int x){for(int i=x;i<=n;i+=lowbit(i)) c[i]+=1;
}
int sum(int x,int y){int res=0;for(int i=y;i;i-=lowbit(i)) res+=c[i];for(int i=x-1;i;i-=lowbit(i)) res-=c[i];return res;
}
int main(){scanf("%d",&n);for(int i=1;i<=n;i++) scanf("%d",&a[i]);for(int i=1;i<=n;i++){great[i]=sum(a[i]+1,n); //计算 i 左侧大于a[i]的数量 low[i]=sum(1,a[i]-1); //计算 i 左侧小于a[i]的数量add(a[i]);}ll cnt1=0,cnt2=0;memset(c,0,sizeof c);for(int i=n;i;i--){cnt1+=great[i]*(ll)sum(a[i]+1,n);cnt2+=low[i]*(ll)sum(1,a[i]-1);add(a[i]);}cout<<cnt1<<" "<<cnt2;return 0;
}