51Nod 1019 - 逆序数（BIT）

1019 逆序数

基准时间限制：1 秒 空间限制：131072 KB 分值: 0 难度：基础题

在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。

如2 4 3 1中，2 1，4 3，4 1，3 1是逆序，逆序数是4。给出一个整数序列，求该序列的逆序数。

Input

第1行：N，N为序列的长度（n <= 50000)

第2 - N + 1行：序列中的元素（0 <= A[i] <= 10^9）

Output

输出逆序数

Input示例

4

2

4

3

1

Output示例

4

 

 

【思路】

   暴力搜索复杂度O(n^2)会超时，这道题的难点在于数据的离散化和用合适的数据结构做优化。首先利用排序加二分搜索将数据离散化，之后把各个数据对应数轴上的点，我们给数轴上的每一个点都加一个相应的标记，初始化都是0，当数据x加入时lable[x]变为1，同时计算出lable[x+1]到lable[maxn]中1的个数，也就是对该区间求和，这样就算出了数据x加入后产生的逆序对数，依次对每一项计算求和边可得到最终的答案。其中lable可用线段树或树状数组实现，这样可以降低时间复杂度。

 

代码一（线段树）

#include<bits/stdc++.h>
using namespace std;

const int maxn = 50050;

#define node tree[id]
#define lson tree[id*2]
#define rson tree[id*2+1]

int n;
int a[maxn], c[maxn];

struct Tree {
	int left, right, value;
}tree[maxn*4];

void pushup(int id) {
	node.value = lson.value + rson.value;
}

void build(int id, int le, int ri) {
	node.left = le;
	node.right = ri;
	node.value = 0;
	if (le == ri) return;
	int mid = (le + ri) >> 1;
	build(id*2, le, mid);
	build(id*2+1, mid + 1, ri);
}

int query(int id, int x, int y) {
	if (x <= node.left && node.right <= y) {
		return node.value;
	}
	int ans = 0;
	int mid = (node.left + node.right) >> 1;
	if (x <= mid) ans += query(id*2, x, y);
	if (y > mid) ans += query(id*2 + 1, x, y);
	return ans;
}

void update(int id, int pos) {
	if (node.left == node.right) {
		node.value = 1;
		return;
	}
	int mid = (node.left + node.right) >> 1;
	if (pos <= mid) update(id*2, pos);
	else update(id*2+1, pos);
	pushup(id);
}

int main() {
	while (scanf("%d", &n) == 1) {
		for (int i = 1; i <= n; i++) {
			scanf("%d", &a[i]);
			c[i] = a[i];
		}
		sort(c + 1, c + 1 + n);
		for (int i = 1; i <= n; i++) {
			a[i] = lower_bound(c + 1, c + 1 + n, a[i]) - c;
		}
		
		/*for (int i = 1; i <= n; i++) {
			cout << a[i] << ' ';
		}
		cout << endl;*/
		
		build(1, 1, n);
		int ans = 0;
		for (int i = 1; i <= n; i++) {
			if (a[i] + 1 <= n)	ans += query(1, a[i] + 1, n);
			update(1, a[i]);
		}
		printf("%d\n", ans);
	}
	return 0;
}

代码二（树状数组）

#include<bits/stdc++.h>
using namespace std;

const int maxn = 50500;

int n;
int a[maxn], c[maxn];
int bit[maxn];

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

int sum(int i) {
	int ans = 0;
	while (i > 0) {
		ans += bit[i];
		i -= lowbit(i);
	}
	return ans;
}

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

int main() {
	while (scanf("%d", &n) == 1) {
		memset(bit, 0, sizeof(bit));
		for (int i = 1; i <= n; i++) {
			scanf("%d", &a[i]);	
			c[i] = a[i];
		}
		sort(c + 1, c + 1 + n);
		for (int i = 1; i <= n; i++) {
			a[i] = lower_bound(c + 1, c + 1 + n, a[i]) - c;
		}
		
		int ans = 0;
		for (int i = 1; i <= n; i++) {
			ans += sum(n) - sum(a[i]);
			add(a[i], 1);
		}
		printf("%d\n", ans);
	}
	return 0;
}