题目大意:在数组中找出三个数,使中间的那个数小于两边的数,问有多少集合。
思路:对于每个数来说,它可以组成的组数就是其左边比之小的数乘上右边比之大的数加上右边比之小的数的乘上左边比之大的数。而这些数据可以通过权值树状数组获得
AC代码:
//LA4329
#include<iostream>
#include<cstdio>
#include<cstring>
#define lowbit(x) x&(-x)
using namespace std;
const int size=20005;
int s[200000];
void add(int x)
{
while(x<=100000)
{
++s[x];
x+=lowbit(x);
}
}
int query(int x)
{
int sum=0;
while(x>=1)
{
sum+=s[x];
x-=lowbit(x);
}
return sum;
}
int c[size],d[size];
int val[size];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&val[i]);
}
memset(s,0,sizeof(s));
for(int i=1;i<=n;i++)
{
c[i]=query(val[i]);
add(val[i]);
}
memset(s,0,sizeof(s));
for(int i=n;i>=1;i--)
{
d[i]=query(val[i]);
add(val[i]);
}
long long sum=0;
for(int i=1;i<=n;i++)
{
sum+=c[i]*(n-i-d[i])+(i-1-c[i])*d[i];
}
cout<<sum<<endl;
}
}