思路1
最简单的方法是顺序数组,将每个数字与后面的比较,统计逆序对的个数,这种方法的时间复杂度为O(n*n)
,这种方法会超时。这里可以巧妙利用归并排序的思想
,只需在归并排序的基础上稍作修改即可!
我们以数组{10,5,6,8,4,7,9}为例来分析统计逆序对的过程,每次扫描到一个数字的时候,我们不能拿它和后面的每一个数字做比较,否则时间复杂度就是O(n*n)
因此我们可以考虑先比较两个相邻的数字。
如下图所示,我们根据归并数组分解成两个长度分别为3和4的子数组,再把这两个子数组分别分解成两个子数组。接下来一边合并相邻的子数组,一边统计逆序对的数目
。在第一对长度为1的子数组{5},{6}中5小于6,因此不组成逆序对。而在第二对长度为1的子数组{8},{4}中8>4组成逆序对{8,4},因此count+1。由于我们已经统计了子数组内部逆序对,因此需要把这子数组排序,以免在以后的统计过程中再重复统计。然后接着合并子数组然后统计逆序对数。
那么如何统计两个长度大于1的子数组之间的逆序对呢?
我们先用两个指针分别指向两个子数组的末尾,并每次比较两个指针指向的数字。如果第一个子数组中的数字大于第二个子数组中的数字,则构成逆序对,并且逆序对的数目等于第二个子数组中的剩余数字的个数
。如果第一个数组中的数字小于或等于第二个数组中的数字,则不构成逆序对。每一次比较的时候,我们都把较大的数字从后往前复制到一个辅助数组中去,确保辅助数组中的数字是递增排序的。在把较大的数字复制到数组之后,把对应的指针向前移动一位,接着来进行下一轮的比较。
经过前面详细的讨论,我们可疑总结出统计逆序对的过程:先把数组分隔成子数组,先统计出子数组内部的逆序对的数目,然后再统计出两个相邻子数组之间的逆序对的数目。在统计逆序对的过程中,还需要对数组进行排序。如果对排序算法很熟悉,我们不难发现这个排序的过程就是归并排序。
AC代码
package com.zhumq.leetcode;
import java.util.ArrayList;
import org.junit.Test;
public class InversePairs {
/*
* 利用归并排序过程统计逆序对
*/
private ArrayList<Integer> assignList(ArrayList<Integer> list ,int start,int end){
ArrayList<Integer> des = new ArrayList<Integer>();
for(int i = start;i<end;i++){
des.add(list.get(i));
}
return des;
}
public long mergeTwoList(ArrayList<Integer> list,int start,int half,int end){
long count = 0;
ArrayList<Integer> tempLeft = assignList(list,start,half);
ArrayList<Integer> tempRight = assignList(list,half,end);
int leftIndex = 0;
int rightIndex = 0;
int index = start;
while(leftIndex < tempLeft.size() && rightIndex <tempRight.size()){
int temp1 = tempLeft.get(leftIndex);
int temp2 = tempRight.get(rightIndex);
if(temp1 > temp2){
System.err.println("["+temp1+","+temp2+"]");
count+=tempLeft.size() - leftIndex;
list.set(index, temp2);
index++;
rightIndex++;
}else{
list.set(index, temp1);
index++;
leftIndex++;
}
}
for(;leftIndex < tempLeft.size();leftIndex++){
list.set(index, tempLeft.get(leftIndex));
index++;
}
for(;rightIndex <tempRight.size();rightIndex++){
list.set(index, tempRight.get(rightIndex));
index++;
}
return count;
}
public long getInversions(ArrayList<Integer> list,int start,int end){
long count = 0;
if((end-start)<= 1)
return 0;
int half = start+(end-start)/2;
count += getInversions(list,start,half);
count += getInversions(list,half,end);
count += mergeTwoList(list,start,half,end);
return count;
}
public long getInversePairs(int[] arr){
ArrayList<Integer> al = new ArrayList<Integer>();
for(int i = 0;i<arr.length;i++){
al.add(arr[i]);
}
int end =arr.length;
return getInversions(al,0,end);
}
@Test
public void test1() {
int[] arr={10,5,6,8,4,7,9};
System.out.println(getInversePairs(arr));
}
}