树状数组求逆序对【TSOJ 1232】

传送门就不传送了,你们也传送不进来;

就是一个求逆序对的裸题;

直接离散化+树状数组就完事了。

树状数组区间更新和区间查询写 这题总是不对;

我也不知道为啥,过几天再研究吧。

单点更新的树状数组写对了;

总之就是如果用树状数组的话一定要离散化。

下面是代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+7;
int sum[maxn];
int a[maxn],b[maxn];
int n;
void init()
{
	memset(sum,0,sizeof(sum));
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
}
void add(int p,int x)
{
	for(int i=p;i<=n;i+=(i & -i))
	{
		sum[i] += x;
	}
}
int query(int p)
{
	int ans = 0;
	for(int i=p;i;i-=(i & -i))
	{
		ans += sum[i];
	}
	return ans;
}
int solve()
{
	int ans = 0;
	for(int i=0;i<n;i++)
	{
		ans += i-query(a[i]);
		add(a[i],1);
	}
	return ans;
}
int main()
{
	while(cin>>n)
	{
		init();
		for(int i=0;i<n;i++)
		{
			cin>>a[i];
			b[i] = a[i];
		}
		sort(b,b+n);
		int m = unique(b,b+n)-b;
		for(int i=0;i<n;i++)
		{
			a[i] = lower_bound(b,b+m,a[i])-b+1;
		}
		cout<<solve()<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/KIKO_caoyue/article/details/84194128