HDU 2492 Ping pong 树状数组

题字:有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);
	}
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值