uvalive 4329 Ping pong---树状数组

本文链接：https://blog.csdn.net/Calmei/article/details/25343335

Problem ：n（3<=n<=20000) 个乒乓球爱好者，每个人有一个技能值，他们的每场比赛位两个选手一个裁判，裁判必须住在两个选手之间，并且技能值也在两个选手之间（比一个小比另一个大），一共能组织多少种比赛。

Way ：如果第i个人当裁判，假设a1到a(i-1)有lelt[i]个比ai小，那么就有(i-1)-left[i]个比a[i]大，同理a(i+1)到a[n]，有right[i]个比a[i]小，就有(n-i)-right[i]个比a[i]大，则第i个人当裁判有：left[i]*(n-i-right[i])+(i-left[i]-1)*right[i]种比赛，只需

求出left[i]和right[i]即可，x[i] = 1表示已有a[i]，详见代码：

code:

//
//  main.cpp
//  Calmeii
//  树状数组
//  Created by Calmeii on 14-5-7.
//  Copyright (c) 2014年 apple. All rights reserved.
//
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define MAXN 20010  //数字最大个数
#define lowbit(v) ((v)&(-v))
// a存放数值，r,l分别存放右边和左边比i小的个数,x存放前缀和，mx最大的那个数
int a[MAXN], r[MAXN], l[MAXN], n, x[100010], mx;
int sum(int id)
{
    int ans = 0;
    while (id > 0)
    {
        ans += x[id];
        id -= lowbit(id);
    }
    return ans;
}
void add(int id, int v)
{
    while (id <= mx)
    {
        x[id] += v;
        id += lowbit(id);
    }
}
int main()
{
    int t;
    scanf("%d", &t);
    while (t--)
    {
        int n;
        mx = 0;
        scanf("%d", &n);
        for (int i = 1; i <= n; i++){
            scanf("%d", &a[i]);
            mx = max(mx, a[i]);
        }
        memset(x, 0, sizeof(x));    //x的初始化
        for (int i = 1; i <= n; i++){
            add(a[i],1);    //前缀和加
            l[i] = sum(a[i]-1);   //i左边的比a[i]小的个数
        }
        memset(x, 0, sizeof(x));    //x的初始化
        for (int i = n; i >= 1; i--){
            add(a[i], 1);
            r[i] = sum(a[i]-1);     //i右边的比a[i]小的个数
        }
        long long ans = 0;
        for (int i = 1; i <= n; i++){
            ans += l[i]*(n-i-r[i]) + (i-l[i]-1)*r[i];
        }
        printf("%lld\n", ans);
    }
    
    return 0;
}
/*
2
6
3 2 8 4 9 2
*/