题意:
一条大街上住着n个乒乓球爱好者,经常组织比赛切磋技术。每个人都有一个不同的技能值a[i]。每场比赛需要3个人,两名选手,一名裁判。他们有一个奇怪的规定,既裁判必须住在两名选手的中间,并且技能值也在两名选手直接。问一共能组织多少种比赛。 1<=T<=20, 3<=n<=20000, 1<=a[i]<=100000。
分析
假设c[i]为 a[1]到a[i-1]中比a[i]小的数量,d[i]为a[i+1]到a[n]中比a[i]小的数量。枚举2到n-1作为裁判,对于每个裁判i,都有c[i]*(n-i-d[i])+d[i]*(i-c[i]-1)种比赛方案。
用求逆序对的方法求出c和d的数组。而且因为a[i]<=100000,所以也不需要进行离散化。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=20000+50;
const int maxa=100000+100;
int T,n,r;
int a[maxn],sumv[4*maxa] ;
int c[maxn],d[maxn];
int p,v;
void update(int o,int L,int R){
if(L==R){
sumv[o]+=v;
return ;
}
int M=L+(R-L)/2;
if(p<=M)update(2*o,L,M);
if(p>M)update(2*o+1,M+1,R);
sumv[o]=sumv[2*o+1]+sumv[2*o];
}
int ql,qr,_sum;
void query(int o,int L,int R){
if(ql<=L&&qr>=R){
_sum+=sumv[o];
return ;
}
int M=L+(R-L)/2;
if(ql<=M)query(2*o,L,M);
if(qr>M)query(2*o+1,M+1,R);
}
int main(){
scanf("%d",&T);
for(int k=1;k<=T;k++){
scanf("%d",&n);
for(int i=1;i<=n;i++){
cin>>a[i];
r=max(r,a[i]);
}
memset(sumv,0,sizeof(sumv));
for(int i=1;i<=n;i++){
p=a[i];v=1;
update(1,1,r);
_sum=0;ql=1;qr=a[i];
query(1,1,r);
c[i]=_sum-1;
}
memset(sumv,0,sizeof(sumv));
for(int i=n;i>=1;i--){
p=a[i];v=1;
update(1,1,r);
_sum=0;ql=1;qr=a[i];
query(1,1,r);
d[i]=_sum-1;
}
long long ans=0;
for(int i=2;i<n;i++){
ans+=c[i]*(n-i-d[i])+(i-1-c[i])*d[i];
}
cout<<ans<<endl;
}
return 0;
}