HDU 2492 Ping pong (树状数组)

最新推荐文章于 2019-09-27 21:03:01 发布

weixin_30487701

最新推荐文章于 2019-09-27 21:03:01 发布

阅读量71

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jxgapyw/p/4749966.html

本文介绍了一种算法，用于计算特定条件下可能的比赛组合数量。具体来说，对于一条街道上的n个乒乓球爱好者，每场比赛由两名选手和一名裁判组成，裁判位于两名选手之间且技能值不低于选手。文章详细解释了如何使用树状数组来快速找到符合条件的比赛组合。

题目大意: 一条大街上住着 n 个乒乓球爱好者，经常比赛切磋技术，每个人有一个不同的技能值ai。每场比赛需要三个人，两名选手，一名裁判。但是裁判必须住在两名选手之间，而且裁判的技能值不能小于或小于两名选手的技能值。问共有多少种比赛。

思路 :枚举裁判，找出位于裁判前面的比裁判技能小的人数和位于裁判后面技能比裁判大的人数，这两个数相乘，并累加到sum中；再找出位于裁判前面技能比裁判大的人数，和位于裁判后面的技能比裁判小的人数，这两个数相乘，将积累加到sum中。

找人数时要用到树状数组：譬如找位于裁判前面技能比裁判小的人数时，用 pre[i] 存储位于裁判前面的技能为 i 的人数，故位于裁判 (技能为a[k]) 前面的技能比裁判小的人数为pre[1] + per[2] + ... + pre[a[k]] 的累加和。其它的情况可以此类推。

  #include<iostream>
  #include<cstdio>
  #include<cstring>
  #include<cstdlib>
  #include<cmath>
  #include<queue>
  #include<map>
  #include<algorithm>
  using namespace std;
  #define INF 0x7fffffff
  int n,t,a[100002],pre[100002],late[100002];
  long long sum,c1[100002],c2[100002],MAX; 

  int lowbit(int x){
return x & (-x);
  }

  int SUM(int x,long long c[]){
int res = 0;
while(x > 0){
    res += c[x] ;
    x -= lowbit(x) ;
}
return res;
  } 

  void add(long long c[],int x,int d){
int t = x;
int p ;
while(t<=MAX){
    c[t] += d;
    t = t + lowbit(t); 
} 
  }
  int main(){
int i,j;
scanf("%d",&t);
while(t--){
    sum = 0;
    MAX = 0 ;
    scanf("%d",&n); 
    for(i=1;i<=n;i++){
        scanf("%d",&a[i]);
        MAX = MAX > a[i] ? MAX : a[i] ; 
    }
    memset(pre,0,sizeof(pre));
    pre[a[1]] = 1;
    memset(late,0,sizeof(late));
    for(i=2;i<=n;i++)
        late[a[i]] = 1;
    memset(c1,0,sizeof(c1));
    memset(c2,0,sizeof(c2));            
    for(i=1;i<=100000;i++){
        for(j=i - lowbit(i) + 1; j<=i; j++){
            c1[i] += pre[j] ;
            c2[i] += late[j] ; 
        }       
    }
    for(i=2;i<n;i++){
        add(c2,a[i],-1);
        sum += SUM(a[i]-1,c1) * (SUM(MAX,c2) - SUM(a[i],c2)) ;
        sum += (SUM(MAX,c1) - SUM(a[i],c1)) * SUM(a[i]-1,c2) ;
        add(c1,a[i],1);
    }
    printf("%I64d\n",sum);
   }
   return 0;
  }

转载于:https://www.cnblogs.com/jxgapyw/p/4749966.html