树状数组求逆序对的方法(学习总结)

逆序对:i<j,a[i]>a[j]
树状数组求逆序对:先把tree置0,每次输入一个数x,用1去更新tree
例如:
tree:0 0 0 0 0
x=4:
tree: 0 0 0 0 1
然后对x进行查询read(x),可以得到x之前插入了多少个数,因为x之前插入的数都比x小,所以用当前插入的数的总数i减去read(x)可以得到当前插入的i个数中有多少个比x大,即x有多个逆序对。
离散化: 如果插入的x很大,每次对x进行查询,时间用的多,且tree的空间不够。这时候需要进行离散化。具体做法是每次输入一个数的同时记录下该数输入的顺序是多少。然后对输入的数进行排序。
例如:
输入:9 1 0 5 4
id: 1 2 3 4 5

排序:0 1 4 5 9
id1: 3 2 5 4 1
3 2 5 4 1 和9 1 0 5 4 的逆序对都是6

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) (x)&(-x)
const int maxn=1e5;
using namespace std;
ll n,tree[maxn];
struct node
{
	int val;
	int id;
}arr[maxn];
bool cmp(node a,node b)
{
	return a.val<b.val;
}
void update(int k,int a)
{
	while(k<=n)
	{
		tree[k]+=a;
		k+=lowbit(k);
	}
}
ll read(ll k)
{
	ll sum=0;
	while(k)
	{
		sum+=tree[k];
		k-=lowbit(k);
	}
	return sum;
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>arr[i].val;
		arr[i].id=i;      
	}
	sort(arr+1,arr+n+1,cmp);
	ll ans=0;
	for(int i=1;i<=n;i++)
	{
		update(arr[i].id,1);
		ans+=(i-read(arr[i].id));
		//cout<<arr[i].id<<" "; 
	}
	  cout<<ans<<endl;
	return 0;
} 
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值