LA4329 - Ping pong

ACMega

于 2013-10-24 17:15:04 发布

阅读量364

点赞数

CC 4.0 BY-SA版权

分类专栏： LA 树状数组

本文链接：https://blog.youkuaiyun.com/acmega/article/details/13001929

LA 同时被 2 个专栏收录

7 篇文章

订阅专栏

树状数组

3 篇文章

订阅专栏

本文深入探讨了树状数组的基础概念、实现方法及其应用，包括如何使用树状数组进行区间求和操作，并通过实例展示了其高效性。文章还详细介绍了树状数组的关键函数和优化技巧，帮助读者掌握这一经典数据结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简单的树状数组

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <map>
#define MAXN 100010
using namespace std;
int N = MAXN, C[100010], a[20010], _c[20010], _d[20010];
int lowbit(int x)
{
	return x & (-x);
}
int sum(int x)
{
	int ret = 0;
	while (x > 0)
	{
		ret += C[x];
		x -= lowbit(x);
	}
	return ret;
}
void add(int x, int d)
{
	while (x <= N)
	{
		C[x] += d;
		x += lowbit(x);
	}
}
int main()
{
	int t;
	scanf("%d", &t);
	while (t --)
	{
		int n;
		scanf("%d", &n);
		for (int i = 0; i < n; ++ i)
		{
			scanf("%d", &a[i]);
		}
		memset(C, 0, sizeof(C));
		for (int i = 0; i < n; ++ i)
		{
			add(a[i], 1);
			_c[i] = sum(a[i]) - 1;
		}
		memset(C, 0, sizeof(C));
		for (int i = n - 1; i >= 0; -- i)
		{
			add(a[i], 1);
			_d[i] = sum(a[i]) - 1;
		}
		long long res = 0;
		//for (int i = 0; i < n; ++ i) printf("%d ", _c[i]); puts("");
		//for (int i = 0; i < n; ++ i) printf("%d ", _d[i]); puts("");
		for (int i = 0; i < n; ++ i)
		{
			res += _c[i] * (n - i - _d[i] - 1) + (i - _c[i]) * _d[i];
		}
		printf("%lld\n", res);
	}
}