树状数组求逆序对个数

#include<cstdio>
#include<algorithm>
const int N = 1e6 + 1;
int n, t[N];
namespace lsh{
	int lsh[N], tmp[N];
	void init() {
		for (int i = 1; i <= n; ++i)
			tmp[i] = lsh[i];
		std::sort(tmp + 1, tmp + n + 1);
		auto end = std::unique(tmp + 1, tmp + n + 1);
		for (int i = 1; i <= n; ++i) 
			lsh[i] = std::lower_bound(tmp + 1, end, lsh[i]) - tmp;
	}
}
inline int lowbit(int x) { return x & -x; }
void add(int p) { for (; p <= n; p += lowbit(p))++t[p]; }
int sum(int p) { int s = 0; for (; p; p -= lowbit(p))s += t[p]; return s; }
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) 
		scanf("%d", lsh::lsh+i);
	lsh::init();
	int ans = 0;
	for (int i = 1; i <= n; ++i) {
		add(lsh::lsh[i]);
		ans += i - sum(lsh::lsh[i]);
	}
	printf("%d\n", ans);
}
posted @ 2021-04-10 11:09  _dwt  阅读(52)  评论(0)    收藏  举报