设A[1…n]是一个包含n个不同数的数组。如果在i<j的情况下,有A[i]>A[j],则(i,j)就称为A中的一个逆序对(inversion)【《算法导论》2-4】
现给出一个数列,求该数列中的逆序对数(逆序数)。本节给出三种方法:方法一是最直接的暴力方法;方法二是基于归并分治的思想;方法三是基于线段树的。
【解法一】
暴力方法最直接,也最容易想到,两层for循环就可以算出来逆序数:每遇到一个元素回头遍历寻找比其大的元素个数即可,当然向后寻找比其小的元素个数也可以,复杂度为O(n^2),代码:
- int sum = 0;
- for(int i = 0; i < size; ++i)
- {
- for(int j = i+1; j < size; ++j)
- {
- if(arr[i] > arr[j])
- {
- ++sum;
- }
- }
- }
- return sum;
【解法二】
这种方法最初见于《算法导论》,这里先附上《算法导论》2-4关于逆序对的几点讨论:
1) 如果数组的元素取自集合[1,2,…,n],那么,怎样的数组含有最多的逆序对?它包含多少个逆序对?
析:很容易想到,当一个数列为逆序时,即[n,n-1,n-2,…,1],逆序数是最多的,是多少?当n=1,inversion=0;n=2,inversion在原来基础上增加1;n=3,inversion增加2…n=n时,inversion增加n-1,所以总的inversion为等差求和n(n-1)/2;也可以这样理解,数列[n,n-1,n-2,…,1]中任取两个数皆为逆序对,故逆序数为Cn2,即n(n-1)/2。
2) 插入排序的运行时间与输入数组中逆序对的数量之间有怎样的关系?
析:我们知道插入排序的程序如下
- for(int i = 1; i < n; ++i)
- {
- int key = arr[i];
- for(int j = i; j > 0 && arr[j-1] > key; --j)
- {
- arr[j] = arr[j-1];
- }
- arr[j] = key;
- }
可以看出,每次内层循环都尝试将当前元素放在“合适”的位置上,所以对于内层循环的每一次迭代,都是在消除当前元素相对之前元素所对应的一系列逆序对,故总的循环便是消除数列所有的逆序对;
3) 给出一个算法,它能用O(nlgn)的最坏运行时间,确定n个元素的任何排列中逆序对的数目。(提示:修改合并排序)
《算法导论》在这里直接提示修改合并排序,所以自然就是用到分治归并的思想,那么逆序对是出现在什么时候?出现多少呢?
我们知道归并排序是通过分治,然后归并两个有序的序列成一个有序序列。现假设两段有序序列分别是[beg,mid]和[mid+1,end],在归并过程中(i,j分别为两段序列的下标),如果a[i]<a[j],则不会产生逆序对;但当a[i]>a[j]时,就出现逆序对了,出现了多少?既然[beg,mid]是有序的,那么[i-mid]序列就都能与a[j]构成逆序对,故:mid-i+1
复杂度O(nlgn),代码如下:
- #include<iostream>
- using namespace std;
- /* 归并求逆序对数, arr 存储最终有序结果
- * 在函数外申请一个临时数组作为参数传入
- * 避免递归不断创建临时数组的开销
- */
- int Merge(int * arr, int beg, int mid, int end, int * tmp_arr)
- {
- memcpy(tmp_arr+beg,arr+beg,sizeof(int)*(end-beg+1));
- int i = beg;
- int j = mid + 1;
- int k = beg;
- int inversion = 0; // 合并过程中的逆序数
- while(i <= mid && j <= end)
- {
- if(tmp_arr[i] <= tmp_arr[j])
- {
- arr[k++] = tmp_arr[i++];
- }else
- {
- arr[k++] = tmp_arr[j++];
- inversion += (mid - i + 1);
- }
- }
- while(i <= mid)
- {
- arr[k++] = tmp_arr[i++];
- }
- while(j <= end)
- {
- arr[k++] = tmp_arr[j++];
- }
- return inversion;
- }
- int MergeInversion(int * arr, int beg, int end, int * tmp_arr)
- {
- int inversions = 0; // 记录倒序数
- if(beg < end)
- {
- int mid = (beg + end) >> 1;
- inversions += MergeInversion(arr, beg, mid, tmp_arr);
- inversions += MergeInversion(arr, mid+1, end, tmp_arr);
- inversions += Merge(arr, beg, mid, end, tmp_arr);
- }
- return inversions;
- }
- /* 测试序列 :answer: 10 */
- int testPoint[10] = {
- 1, 4, 2, 9, 48,
- 15, 13, 44, 6, 90
- };
- void main()
- {
- int arrcopy[10]; // 临时数组
- memcpy(arrcopy,testPoint,sizeof testPoint);
- printf("the num of inversions is: %d\n",
- MergeInversion(testPoint,0,9,arrcopy));
- }
一般来讲,在解决数列的区间问题上,线段树有很高的出镜率。逆序数便是一个“区间和”的问题:对于数列中的每个元素,它对应的逆序数便是之前序列中大于该元素的元素个数和。
有了这样的分析,便可以用线段树来解决了:对于数列中每一个元素x,它必将对应这线段树中的一条长logn的路径,该路径中的每条线段都包含x,故每遍历到一个元素则将包含该元素的路径中的每条线段次数都加1,而查询该元素所对应的逆序数便是查询线段树中[x,up]的区间和,其中up为该序列的上界。
举个例子:序列:5,4,3,2,1的逆序数,设up为10
i=0,arr[i]=5,查询区间[5,10]的元素个数,此时线段树空,故inversion=0;将5插入包含它的线段路径。
i=1,arr[i]=4,查询区间[4,10]的元素个数,为1,故inversion+=1;将4插入包含它的线段路径。
i=2,arr[i]=3,查询区间[3,10]的元素个数,为2,故inversion+=0;将3插入包含它的线段路径。
以此类推。。。代码如下:
- #include<iostream>
- using namespace std;
- /* 线段树求逆序数 */
- #define BORDER 100 // 设测试数列最大不超过100
- struct Node // 线段树
- {
- int left;
- int right;
- int counter;
- }segTree[4*BORDER];
- /* 构建线段树 根节点开始构建区间[lef,rig]的线段树*/
- void construct(int index, int lef, int rig)
- {
- segTree[index].left = lef;
- segTree[index].right = rig;
- if(lef == rig) // 叶节点
- {
- segTree[index].counter = 0;
- return;
- }
- int mid = (lef+rig) >> 1;
- construct((index<<1)+1, lef, mid);
- construct((index<<1)+2, mid+1, rig);
- segTree[index].counter = 0;
- }
- /* 包含x的区间的值都+1 即对包含x的logn个节点的路径进行更新*/
- void insert(int index, int x)
- {
- ++segTree[index].counter;
- if(segTree[index].left == segTree[index].right)
- {
- return;
- }
- int mid = (segTree[index].left + segTree[index].right) >> 1;
- if(x <= mid)
- {
- insert((index<<1)+1, x);
- }else
- {
- insert((index<<1)+2, x);
- }
- }
- /* 查询点x的逆序数,即查询[x,BORDER]的区间和 */
- int query(int index, int left, int right)
- {
- if(segTree[index].left == left && segTree[index].right == right)
- {
- return segTree[index].counter;
- }
- int mid = (segTree[index].left+segTree[index].right) >> 1;
- if(right <= mid)
- {
- return query((index<<1)+1,left,right);
- }else if(left > mid)
- {
- return query((index<<1)+2,left,right);
- }
- return query((index<<1)+1,left,mid) + query((index<<1)+2,mid+1,right);
- }
- /* 测试序列 :answer: 10 */
- int testPoint[10] = {
- 1, 4, 2, 9, 48,
- 15, 13, 44, 6, 90
- };
- void main()
- {
- construct(0,0,100); // 构建[0,100]线段树
- int reverseSum = 0; // 记录逆序对数
- for(int i = 0; i < 10; ++i) // 查询当前逆序数
- {
- reverseSum += query(0,testPoint[i],BORDER);
- printf("num of reverse order is: %d\n", reverseSum);
- insert(0,testPoint[i]);
- }
- }
由于线段树的插入和查询操作皆可以在lgn的时间内完成,故遍历一个数列求逆序数的时间复杂度为O(nlgn),空间复杂度为O(nlgn)。