逆序对的求法(树状数组)

设A[1...n]是一个包含n个不同数的数组,如果在i<j的情况下,有A[i]>A[j],则(i,j)就称为A中的一个逆序对。

求法有两种,一种是归并排序,一种是树状数组。二者的时间复杂度都是O(n*logn),但树状数组更加好写。


思路:开一个树状数组A记录每一个数出现的次数。倒序输入,每输入一个数num就在A中加1,同时求一下从0到num-1的前缀和。


#include <iostream>
using namespace std;

void AddNum(int * arr, int iArrSize, int iIndex, int iNum)
{
	for (int i = iIndex; i <= iArrSize; i += (i & (-i)))
		arr[i] += iNum;
}

int GetSum(int * arr, int iIndex)
{
	int iAns = 0;
	for (int i = iIndex; i; i -= (i & (-i)))
		iAns += arr[i];
	return iAns;
}

int main()
{
	int iNumTot, arrSum[1010] = { 0 }, arrNum[1010] = { 0 };
	cin >> iNumTot;
	
	int iAns = 0;
	for (int i = 1; i <= iNumTot; i++)
		cin >> arrNum[i];
		
	for (int i = iNumTot; i >= 1; i--)
	{
		AddNum(arrSum, 1000, arrNum[i], 1);
		iAns += GetSum(arrSum, arrNum[i] - 1);
	}
	
	cout << iAns << endl;
	
	
	return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值