算法基础---逆序对数问题

求逆序对数问题

问题描述

对于数列的第 i 个和第 j 个元素，如果满足 i < j 且 a[i] > a[j]，则其为一个逆序对。

问题分析

可以采用分治法中归并排序的思想来解决问题O(nlogn)

将序列从中间分开可以将逆序对分成三种情况：

• 组成逆序对的两个元素都在左边left
• 组成逆序对的两个元素都在右边right
• 组成逆序对的两个元素跨左右两个区间left和right各一个

于是就有

1. 递归计算左边的逆序数
2. 递归计算右边的逆序数
3. 递归计算左右跨左右两边的逆序数
4. 再把他们累加到一起

#include<iostream>
using namespace std;
typedef long long LL;

const int N = 1e5 + 10;
int n;
int q[N], tmp[N];

LL merge_sort(int q[], int l, int r){
	if(l >= r) return 0;
	//将区间划分成两半 
	int mid = l + r >> 1;
	//递归将左右两个子区间的逆序数相加 
	LL res = merge_sort(q, l, mid) + merge_sort(q, mid + 1, r);
	//归并,i和j分别为left和right的起始元素 
	int k = 0, i = l, j = mid + 1;
	while(i <= mid && j <= r){
		if(q[i] <= q[j]) tmp[k ++] = q[i ++];
		else{
			res += mid - i + 1;	//递归将跨左右两个区间的逆序数累加 
			tmp[k ++] = q[j ++];
		}
	}
	//扫尾
	while(i <= mid) tmp[k ++] = q[i ++];
	while(j <= r) tmp[k ++] = q[j ++];
	
	for(i = l, j = 0; i <= r; i ++, j ++) q[i] = tmp[j];
	//返回总的逆序数个数 
	return res;
}

int main(){
	scanf("%d",&n);
	for(int i = 0; i < n; i ++) scanf("%d",&q[i]);
	cout << merge_sort(q, 0, n-1) << endl;
	
	return 0;
}