题意:给一个长度n的序列A,问有多少四元组(a,b,c,d)满足:4个数两两不同,1 <= a < b <= n,1 <= c < d <= n,A_a < A_b,A_c > A_d。
思路:也就是找一个四元组里,一对顺序对,一对逆序对,问满足这样的四元组有多少 。
找顺序对,逆序对的过程经典树状数组问题O(nlogn),这题点范围需要离散化。
直接统计整个序列顺序对个数乘以逆序对个数会将不满足情况计数,四元组(a,b,c,d)主要有四类情况
a,c a,d b,c b,d相同,减去这四类重复的计数,
我们就需要统计当前位置i,前面比A[i]大pre_max,小的个数pre_min,后面比A[i]大post_max,小的个数post_min。
拿a,c相同为例,a对于b来说的情况个数是pre_min,c相对d个数是pre_max
ans-=pre_max*pre_min;
其他三类情况同理。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=50000+10;
int tree[N],a[N],b[N];
ll pre_max[N],pre_min[N],post_max[N],post_min[N];
int lowbit(int x){
return x&(-x);
}
void update(int x){
while(x<=N){
tree[x]+=1;
x+=lowbit(x);
}
}
int get_sum(int x){
int sum=0;
while(x>0){
sum+=tree[x];
x-=lowbit(x);
}
return sum;
}
int main()
{
int n;
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
b[i]=a[i];
}
sort(b+1,b+n+1);
int m=unique(b+1,b+1+n)-b;
for(int i=1;i<=n;i++){
a[i]=lower_bound(b+1,b+1+m,a[i])-b;
}
memset(tree,0,sizeof(tree));
ll t1=0,t2=0;
for(int i=1;i<=n;i++){
update(a[i]);
pre_min[i]=get_sum(a[i]-1);
pre_max[i]=i-get_sum(a[i]);
}
for(int i=1;i<=n;i++){
post_min[i]=get_sum(a[i]-1)-pre_min[i];
post_max[i]=n-get_sum(a[i])-pre_max[i];
t1+=post_max[i];
t2+=post_min[i];
}
ll ans=t1*t2;
for(int i=1;i<=n;i++){
ans-=pre_min[i]*pre_max[i];
ans-=pre_min[i]*post_min[i];
ans-=post_max[i]*pre_max[i];
ans-=post_max[i]*post_min[i];
}
printf("%lld\n",ans);
}
}