用归并排序求逆序对:

学习笔记,仅供参考,若有错误,还请指正。

题目:

在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。

输入:

第一行输入一个整数 N (1 ≤ N ≤ 10^6),第二行依次输入 N 个整数表示数组中的a1​,a2​,…,aN​.

输出:

输出一个整数 K 表示逆序对的总数。

代码:

#include 
using namespace std;
const int N = 1000010;
int n;
long long cnt = 0;
int q[N],tmp[N];
void f(int q[],int l,int r)
{
	if(l >= r)	return;
	int mid = l+r>>1;
	f(q,l,mid);
	f(q,mid+1,r);
	int k = 0,i = l,j = mid+1;
	while(i<=mid && j<=r){
		if(q[i] <= q[j])	
			tmp[k++] = q[i++];
		else{
			tmp[k++] = q[j++];
			cnt += mid-i+1;
		}	
	}
	while(i <= mid)
		tmp[k++] = q[i++];
	while(j <= r)	
		tmp[k++] = q[j++];
	for(int i=0;i < k;i ++)
		q[i+l] = tmp[i];
}
int main()
{
	cin >> n;
	for(int i=0;i < n;i ++)
		cin >> q[i];
	f(q,0,n-1);
	cout << cnt;
	return 0;
}

你可能感兴趣的:(排序算法)