题字:有n个人要进行乒乓球比赛每一个人都一个技术值,现在要求进行一场比赛,三个人,裁判的技术值在两个人的中间,位置也在两个人的中间,问一共可以进行这种比赛多少次。
思路:
采用树状数组,先从左到右计算左边大的个数,左边小的个数,再从右到做计算右边大和小的个数,然后交叉相乘取和就可以了
注意:数据较大建议使用long long
AC代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define INF 1e9+7
#define N 1000000
int a[N], c[N], ans1[N], ans2[N], ans1x[N], ans2x[N];
long long ans;
int lowbit(int x)
{
return x&(-x);
}
void add(int i, int d)
{
while (i <= N)
{
c[i] += d;
i += lowbit(i);
}
}
int sum(int i)
{
int res = 0;
while (i > 0)
{
res += c[i];
i -= lowbit(i);
}
return res;
}
int main()
{
int t, n, i, j, k;
scanf("%d", &t);
while (t--)
{
scanf("%d", &n);
memset(c, 0, sizeof(c));
for (i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
k = sum(a[i]);
ans1[i] = k; //前i个数中比a[i]小的个数
ans2[i] = i - k - 1; //前i个数中比a[i]大的个数
add(a[i], 1);
}
memset(c, 0, sizeof(c));
for (i = n, j = 1; i >= 1; i--, j++)
{
k = sum(a[i]);
ans1x[i] = k; //后j个数中比a[i]小的个数
ans2x[i] = j - k - 1; //后j个数中比a[i]大的个数
add(a[i], 1);
}
for (i = 1, ans = 0; i <= n; i++)
{
ans += ans1[i] * ans2x[i] + ans2[i] * ans1x[i]; //交叉相乘积
}
printf("%lld\n", ans);
}
}