HDU2491 Priest John's Busiest Day

題目鏈接

題意：

有n個人要進行乒乓球比賽，每個人都一個能力值，每個人出現的次序就是他們住的位置

現在要求進行一場比賽，三個人，裁判的能力值在兩個選手之間，住的位置也在兩個人的之間

問這種比賽一共可以進行多少次

思路：

用樹狀數組做，否則TLE，先從左到右掃一遍，計算每點左邊大的個數和小的個數，

再從右到左掃一遍，計算每點右邊大和小的個數，然後交叉相乘取和就可以了

代碼如下：

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
int n;
int p[N], c[N], li[N], la[N], ri[N], ra[N];

inline int Lowbit(int x){ return x&(-x); } 

void change(int u, int x)
{
	while(u < N)
	{
		c[u] += x;
		u += Lowbit(u);
	}
} 

int get_sum(int x)
{
	int ans = 0;
	for(int i = x; i > 0; i -= Lowbit(i))
	{
		ans += c[i];
	}
	return ans;
}

int main()
{
	int t;
	scanf("%d", &t);
	while(t--)
	{
		memset(c, 0, sizeof(c));
		scanf("%d", &n);
		for(int i = 1; i <= n; i++)
		{
			scanf("%d", &p[i]);
			int cnt = get_sum(p[i]);
			li[i] = cnt; // i點左邊比它小的 
			la[i] = i - cnt - 1; //i點左邊比它大的 
			change(p[i], 1);
		}
		memset(c, 0, sizeof(c));
		for(int i = n; i > 0; i--)
		{
			int cnt = get_sum(p[i]);
			ri[i] = cnt; // i點右邊比它小的 
			ra[i] = n - i - cnt; //i點右邊比它大的 
			change(p[i], 1); 
		}
		ll ans = 0;
		for(int i = 1; i <= n; i++)
		{
			ans += li[i] * ra[i] + la[i] * ri[i];
		}
		printf("%I64d\n", ans);
	}
	return 0;
}