LA4329 Ping pong(树状数组)

题目链接

一群乒乓爱好者组织比赛，给出一群乒乓爱好者的技能值，每场比赛需要两名选手和一名裁判。裁判必须在选手中间，并且技能值也在选手中间，问一共能组织多少种比赛？

选手的技能值：A1······Ai······An;

假设在A1到Ai内有c个选手的技能值比Ai小，即有(i-1-c)个选手的技能值比Ai大；

假设在Ai到An内有d个选手的技能值比Ai小，即有(n-i-d)个选手的技能值比Ai大；

所以，总共的方案个数：c*(n-i-d)+(i-1-c)*d;

代码如下：

#include <iostream>#include <stdio.h>#include <string.h>using namespace std;const int N=200005;int n;long long tree[N];int t;int data[N];long long pre[N];long long sub[N];int lowbit(int i){    return i&(-i);}void update(int i,int x){    while(i<=N)    {        tree[i]+=x;        i+=lowbit(i);    }}int query(int i){    int ans=0;    while(i>0)    {        ans+=tree[i];        i-=lowbit(i);    }    return ans;}int main(){    scanf("%d",&t);    while(t--)    {        memset(tree,0,sizeof(tree));        memset(pre,0,sizeof(pre));        memset(sub,0,sizeof(sub));        scanf("%d",&n);        for(int i=1;i<=n;i++)            scanf("%d",&data[i]);        for(int i=1;i<=n;i++)        {            pre[i]=query(data[i]-1);            update(data[i],1);        }        memset(tree,0,sizeof(tree));        for(int i=n;i>=1;i--)        {            sub[i]=query(data[i]-1);            update(data[i],1);        }        long long ans=0;        for(int i=1;i<=n;i++)        {            long long c=pre[i];            long long d=sub[i];            ans+=c*(n-i-d)+(i-1-c)*d;        }        printf("%lld\n",ans);    }    return 0;}