POJ 3928 - Ping pong

树状数组的应用。

把每个人当做裁判进行遍历，计算住在这个人之前的人中有多少个是比他弱的，记为pre_less[i],多少个人比他强的记为pre_more[i];同样的，住在他后面的人中，比他弱的记为suf_less[i],比他强的记为suf_more[i]

那么，很显然的对于每个裁判judger[i]，可以进行pre_less[i]*suf_more[i]+pre_more[i]+suf_less[i]场比赛，累加起来即得答案。

#include<cstdio>
#define MAX_N 20000+5
int n,a[MAX_N],c[100000+5],MAX_ai;
int pre_more[MAX_N],pre_less[MAX_N],suf_more[MAX_N],suf_less[MAX_N];
long long ans;
int lowbit(int x){return x&(-x);}
void add(int i,int val)
{
    while(i<=MAX_ai)
    {
        c[i]+=val;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=c[i];
        i-=lowbit(i);
    }
    return s;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		MAX_ai=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			if(MAX_ai<a[i]) MAX_ai=a[i];
		}
		for(int i=1;i<=n;i++) pre_more[i]=0,pre_less[i]=0,suf_more[i]=0,suf_less[i]=0;
		for(int i=1;i<=MAX_ai;i++) c[i]=0;
		for(int i=1;i<=n;i++)
		{
			add(a[i],1);
			pre_less[i]=sum(a[i]-1);
			pre_more[i]=i-1-pre_less[i];
		}
		for(int i=1;i<=MAX_ai;i++) c[i]=0;
		for(int i=n;i>=1;i--)
		{
			add(a[i],1);
			suf_more[i]=sum(MAX_ai)-sum(a[i]);
			suf_less[i]=n-i-suf_more[i];
		}
		ans=0;
		for(int i=1;i<=n;i++)
		{
			ans+=pre_less[i]*suf_more[i]+pre_more[i]*suf_less[i];
		}
		printf("%I64d\n",ans);
	}
}

当然，从代码中不难看出，有两个数组是不必要开的，pre_more[]和suf_less[]是根据pre_less[]和suf_more[]算出来的，那么我们完全可以在最后运算时再计算……

#include<cstdio>
#include<cstring>
#define MAX_N 20000+5
int n,a[MAX_N],c[100000+5],MAX_ai;
int pre_less[MAX_N],suf_more[MAX_N];
int lowbit(int x){return x&(-x);}
void add(int i,int val)
{
    while(i<=MAX_ai)
    {
        c[i]+=val;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=c[i];
        i-=lowbit(i);
    }
    return s;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		MAX_ai=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			if(MAX_ai<a[i]) MAX_ai=a[i];
		}
		for(int i=1;i<=n;i++) pre_less[i]=0,suf_more[i]=0;
		for(int i=1;i<=MAX_ai;i++) c[i]=0;
		for(int i=1;i<=n;i++)
		{
			add(a[i],1);
			pre_less[i]=sum(a[i]-1);
		}
		for(int i=1;i<=MAX_ai;i++) c[i]=0;
		for(int i=n;i>=1;i--)
		{
			add(a[i],1);
			suf_more[i]=sum(MAX_ai)-sum(a[i]);
		}
		long long ans=0;
		for(int i=2;i<=n-1;i++)
		{
			ans+=pre_less[i]*suf_more[i]+(i-1-pre_less[i])*(n-i-suf_more[i]);
		}
		printf("%I64d\n",ans);
	}
}

不难看到，程序在各方面都得到了小幅优化