求逆序数(光影切割问题)

《编程之美》第1.7节，光影切割问题。其中在最后一部分提出了求逆序数的需求，需要在O(nlogn)复杂度下计算逆序数。

算法思想：利用归并排序时左右数组都已经有序了，例如左边1,3,5，右边2,4,6，此时，开始合并，当比较3和2时，由于3>2,因此出现了第一个逆序数，此时，3后面的数都比2大，因此需要增加的逆序数个数就是3后面的元素个数，如果5的下标为mid，3的下标为i，那么逆序数个数num+=mid-i+1。然后开始比较2后面的，每次遇到右边的比左边的数小时，就增加逆序数，当这个数组都排序好了以后逆序数自然也就算出来了，因此复杂度和归并排序的复杂度相同，都为O(nlogn)。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int num=0;

void merge(int A[],int begin,int mid,int end)
{
	int i=begin,j=mid+1;
	int k=begin;
	int *B=new int[mid-begin+1];
	for(int m=0;m<mid-begin+1;m++)
		B[m]=A[begin+m];
	while(i<=mid || j<=end)
	{
		int first=i>mid?INT_MAX:B[i-begin];
		int second=j>end?INT_MAX:A[j];
		if(first<second)
		{
			A[k++]=first;
			i++;
		}
		else
		{
			if(i<=mid)
				num+=mid-i+1;
			A[k++]=second;
			j++;
		}
	}
	delete []B;
}

void mergeSort(int A[],int first,int last)
{
	if(first==last)
		return ;
	int mid=first+((last-first)>>1);
	mergeSort(A,first,mid);
	mergeSort(A,mid+1,last);
	merge(A,first,mid,last);
}

int hundrandRand()
{
	return rand()%100;
}

int main()  
{  
	const int N=10;
	int A[N];
	int temp[N];
	generate(A,A+N,hundrandRand);
	copy(A,A+N,temp);
	mergeSort(A,0,N-1);

    system("pause");  
    return 0;  
}

参考 ：http://blog.csdn.net/dlengong/article/details/7594919

后面还有一个用树状数组求和来解的，有时间做一下。。。

http://www.cppblog.com/Ylemzy/articles/98322.html