[树状数组&线段树]HDU 2492 ping pong

/**
* 一条大街上有n个乒乓球爱好者,经常组织比赛,每个人都有一个排名,
* 每场比赛需要3个人,两名选手和一名裁判。规定裁判位置必须在选手中
* 间,排名也必须在两个选手的中间,问一共能组织多少场比赛
*
* 裁判位置在两个选手中间,而且排名也要在选手中间,那么有两种情况
* 左<裁判<右 或者 右>裁判>左
* 当第i个人当裁判时,假设 a[i] 的左边中有 c[i] 个比a[i]小,那左边就有
* (i-1)-c[i] 个比 a[i] 大。同理 a[i] 的右边中有 d[i] 个比 a[i] 小,
* 那么右边就有 (n-i)-d[i] 个比 a[i] 大,
* 根据乘法原理,从比裁判小和比裁判大的元素中各取出一个,就可以完成统计
* 当i当裁判时,总共有这么多 c[i] * (n-i-d[i]) + (i-c[i]-1) * d[i] 可能

  • 问题转换为求d[i]和c[i],即求第i个人当裁判时,左边比他能力值更小的人数
  • 右边比他能力值更大的人数
    *
  • 对于 c[i],d[i] 所有可能的情况,使用线段树或树状数组 ,均可对数组ci,di,做sum(1~i) 求和
    */
#include <cstdio>
#include<iostream>
#include <algorithm>
#include<cstring>
using namespace std;
int  T,bit[100001],a[20001],c[20001],d[20001],n,mn;
int Sum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=bit[x];
        x-= x&-x;
    }
    return ans;
}
void add(int x,int i)
{
    while(x<=mn)
    {
        bit[x]+= i;
        x+= x&-x;
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        mn=0;
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]);
            mn=max(a[i],mn);
        }
        memset(bit,0,sizeof(bit));
        for(int i = 1; i <= n; i++)
        {
            add(a[i],1); // 第a[i]个人的技能值+1
            c[i] = Sum(a[i]-1);//求出第i个人的前面排名比他小的人数,存入数组c中
        }
        memset(bit,0,sizeof(bit));
        for(int i = n; i >= 1; i--)
        {
            add(a[i],1);
            d[i] = Sum(a[i]-1);//求出第i个人的后面排名比他小的人数,存入数组d中
        }
        long long ans=0;
        for(int i = 2; i < n; i++)
            ans += c[i]*(n-i-d[i]) + (i-c[i]-1)*d[i];
        printf("%I64d\n",ans);
    }

    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值