题意:在一条街上依次住着n个乒乓球爱好者,每个人都有一个不同的能力值ai。每场比赛有两个选手和一个裁判,规定裁判必须住在两个选手中间且能力值同样在两位选手中间。问最多能组织多少场比赛。
题解:考虑第i个人当裁判的情况。假设在a1到ai-1中有ci个比ai小,那么就有i-1-ci个比ai大;假设在ai+1到an中有di个比ai小,那么就有n-i-di个比ai大。此时i当裁判就一共有ci(n-i-di)+di(i-1-ci)种情况。
问题转化为求ci与di。从左到右扫描一遍ai,令x[j]表示从a1到ai-1是否出现过一个ai=j,其中0表示没有,1表示有,然后维护x[j]的前缀和便是ci。用相同方法从右向左扫描得到的就是di。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<stack>
#include<vector>
#include<iostream>
#include<algorithm>
#define maxn 100050
#define INF 0x3f3f3f3f
#define eps 1e-8
const int mod=1e9+7;
using namespace std;
typedef long long ll;
int t,n,maxx;
int a[maxn],c[maxn];
int suml[maxn],sumr[maxn];
int lowbit(int x){return x&(-x);}
void init()
{
maxx=0;
memset(suml,0,sizeof(suml));
memset(sumr,0,sizeof(sumr));
memset(c,0,sizeof(c));
}
int sum(int x)
{
int ans=0;
while(x>0)
{
ans+=c[x];
x-=lowbit(x);
}
return ans;
}
void add(int x,int i)
{
while(x<=maxx)
{
c[x]+=i;
x+=lowbit(x);
}
}
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
init();
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
maxx=max(maxx,a[i]);
}
for(int i=1;i<=n;i++)
{
suml[i]=sum(a[i]);
add(a[i],1);
}
memset(c,0,sizeof(c));
for(int i=n;i>=1;i--)
{
sumr[i]=sum(a[i]);
add(a[i],1);
}
ll ans=0;
for(int i=2;i<n;i++)
ans+=((ll)suml[i]*(n-i-sumr[i])+(ll)sumr[i]*(i-1-suml[i]));
printf("%lld\n",ans);
}
return 0;
}