题目链接:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=13895
题目大意:一条街上住n个乒乓球手,每个人技能值不同,当一个人住在两个人之间,且技能值在两人之间时,这三个人就会打比赛了(中间这个人当裁判),问能打多少场比赛?
分析:当第i个人当裁判时,
假设前a[1]~a[i-1]中有c[i]个人比a[i]小,那么就有(i-1-c[i])个人比a[i]大,同理,假设后a[i+1]~a[n]中有d[i]个比a[i]小,则有(n-i-d[i])个人比a[i]大——>ans=ans+c[i]*(n-i-d[i])+d[i]*(i-1-c[i])
不过,如何算c[i]和d[i]呢?
我们先假设有一个数组x来判断技能值j这个值是否存在(x[i]=1表示存在,x[i]=0表示不存在),然后对于a[i],只要统计x[1]+x[2]+……+x[a[i]-1]的值就可以得出c[i]了,到这里应该很明显了————【树状数组】求前缀和。
同理,我们也可以逆向统计求出d[i]。
如果你不会树状数组,额……我也只能表示没办法。
下附代码:
#include<cstdio>
#include<cstring>
#define maxn 100100
#define mm 20050
int a[mm],s[maxn],c[mm],d[mm];//s表示树状数组,c表示a[i]前面比a[i]小的数量,d则表示后面
int lowbit(int x){return x&-x;}
int sum(int x)
{
int ret=0;
while(x>0)
{
ret+=s[x];
x-=lowbit(x);
}
return ret;
}
void add(int x)
{
while(x<=maxn)
{
s[x]+=1;
x+=lowbit(x);
}
}
int main()
{
int t,n;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(s,0,sizeof(s));
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
{
add(a[i]);
c[i]=sum(a[i])-1;//直接求s[1~i]
}
memset(s,0,sizeof(s));//别忘了清空数组
for(int i=n;i>=1;i--)
{
add(a[i]);
d[i]=sum(a[i])-1;
}
long long ans=0;
for(int i=2;i<n;i++)
ans=ans+c[i]*(n-i-d[i])+d[i]*(i-c[i]-1);
printf("%lld\n",ans);
}
return 0;
}