#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,a[20005],c[20005],d[20005];
int e[200005];
inline int lowbit(int x)
{
return x&(-x);//2^k
}
int sum(int x) //求前 N項的和
{
int r=0;
while(x>0)
{
r+=e[x];
x-=lowbit(x);
}
return r;
}
void add(int x,int d)//求前 N項sum的和
{
while(x<=100000)
{
e[x]+=d;x+=lowbit(x);
}
}
int main()
{
int t,i;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(a,0,sizeof(a));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
memset(e,0,sizeof(e));
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
for(i=1;i<=n;i++)//前綴和
{
add(a[i],1);//求前 N項sum的和
c[i]=sum(a[i]-1);//求前 N項的和
}
memset(e,0,sizeof(e));
for(i=n;i>=1;i--)//後綴和
{
add(a[i],1);
d[i]=sum(a[i]-1);
}
long long ans=0;
for(i=2;i<n;i++)
ans+=(long long)c[i]*(n-i-d[i])+(long long)d[i]*(i-1-c[i]);
printf("%lld\n",ans);
}
return 0;
}
Ping pong
思路: 採用前綴和,後綴和得到結果
再求前綴和,後綴和的時候後用樹狀數組算法得出來。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.