C++ 实现归并排序求逆序数经典算法

在归并排序的基础上,增加一行代码即可

#include<iostream>
#define N 100000
long long res = 0;

void __merge(int arr[], int left, int mid, int right)
{
	int *aux = new int[right - left + 1];
	// 拷贝一遍数组
	for (int i = left; i <= right; i++)
		aux[i - left] = arr[i];

	int i = left, j = mid + 1;
	// arr数组用来排序,aux数组是原来的数组序列
	for (int k = left; k <= right; k++) {
		if (i > mid) { arr[k] = aux[j - left]; j++; }        // 左边已排序完成
		else if (j > right) { arr[k] = aux[i - left]; i++; }    // 右边已排序完成
		else if (aux[i - left] < aux[j - left]) { arr[k] = aux[i - left]; i++; }  // 左边小于右边
		else { // 右边小于左边
			arr[k] = aux[j - left]; 
			res += mid + 1 - i;
			j++; 
		

		}                
	}
	delete[]aux;
}

// 递归使用归并排序,对arr[l...r]的范围进行排序

void __mergeSort(int arr[], int left, int right) {

	if (left >= right)
		return;
	int mid = left + (right - left) / 2;
	__mergeSort(arr, left, mid);
	__mergeSort(arr, mid + 1, right);
	__merge(arr, left, mid, right);
}

int main() 
{
	int n;
	int arr[N];
	std::cin >> n;
	for (int i = 0; i < n; i++)
		std::cin >> arr[i];
	__mergeSort(arr, 0, n - 1);
	std::cout << res << std::endl;

}

 

点赞