pat-top 1009. Triple Inversions (35)

https://www.patest.cn/contests/pat-t-practise/1009

这道题首先用了一个简单的dp：

const int maxn = 1e5 + 2;
int lar[maxn];
int a[maxn];

int main()
{
	int n, k, ans = 0;
	scanf("%d", &n);
	for (int i = 0;  i < n;  i++)
	{
		scanf("%d", &a[i]);
		for (int j = 0; j < i; j++) if (a[i] < a[j]) {
			lar[i]++;
			ans += lar[j];
			//printf("i,j:%d %d a[i],a[j]:%d %d lar[j]:%d ans=%d\n", i, j, a[i], a[j], lar[j], ans);
		}
	
	}
	printf("%d\n", ans);
    return 0;
}

超时。

原因是每次遍历i之前的元素看是否比a[i] 小并计算逆序数太慢了。O(n^2);

需要更好的数据结构平衡查询和插入的时间，使插入和查询的时间都控制在O(logn) 之内。使用树状数组求逆序数。

参考了http://blog.csdn.net/jtjy568805874/article/details/50759553

使用树状数组：统计目前某个元素出现的次数，查询时看比a[i]小的元素出现了几次。

#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 2;

int n, a[maxn], c[maxn], L[maxn], R[maxn];//c 是树状数组中的c数组，L是add从左侧开始以a[i]结尾的逆序数，R是从右侧开始以a[i]为结尾的逆序数 

int lowbit(int x) { return x&-x; }

void add(int i, int x) {
	while (i <= n) {
		c[i] += x;
		i += lowbit(i);
	}
}

int sum(int i) {
	int res = 0;
	while (i) {
		res += c[i];
		i -= lowbit(i);
	}
	return res;
}

int main() 
{
	long long  res = 0;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	for (int i = 1; i <= n; i++) c[i] = 0;
	for (int i = 1; i <= n; i++) { add(a[i], 1); L[i] = i - sum(a[i]); } //add(a[i]，1): a[i] 出现了一次; L[i] = i-sum(a[i]): i之前比a[i] 大的元素有几个
	for (int i = 1; i <= n; i++) c[i] = 0;
	for (int i = n; i >= 1; i--) { R[i] = sum(a[i]); add(a[i], 1);  } //R[i]=sum(a[i]); i之后比a[i]小的元素有几个；
	for (int i = 1; i <= n; i++) res += (long long )L[i] * R[i]; //以a[i] 为中间元素，的triple inversion的个数
	printf("%lld\n", res);
	return 0;
}