HDOJ2838(树状数组)

大意：求所有逆序数对的和

分析：对于数a，他的逆序数对之和为：逆序对数*a+a之前比a大的数。开两个树状数组，一个求逆序对数，一个求和。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 100001
using namespace std;
struct tree
{
	int t;
	long long sum;
}tree[MAXN];
int n;
int lowbit(int i)
{
	return i&(-i);
}
void update(int i, int s, int x)
{
	while (i <= n)
	{
		tree[i].t += x;
		tree[i].sum += s;
		i = i + lowbit(i);
	}
}
int queryx(int n)
{
	int ans = 0;
	while (n > 0)
	{
		ans += tree[n].t;
		n -= lowbit(n);
	}
	return ans;
}
long long  querysum(int n)
{
	long long ans = 0;
	while (n > 0)
	{
		ans += tree[n].sum;
		n -= lowbit(n);
	}
	return ans;
}
int main()
{
	while (~scanf("%d", &n))
	{
		long long ans = 0;
		memset(tree, 0, sizeof(tree));
		for (int i = 1; i <= n; i++)
		{
			int a;
			scanf("%d", &a);
			update(a, a, 1);
			long long k = i - queryx(a);
			if (k != 0)
			{
				long long k1 = querysum(n) - querysum(a);//前n个数之和减去比a小的数之和，即比a大的数之和。
				ans += k*a + k1;
			}
		}
		printf("%lld
", ans);
	}
	return 0;
}