java歸併排序的應用--求逆序數

上一篇我們講了歸併排序的應用和思路,我們現在藉助這個歸併排序的思路,即每個元素要和左邊所有,或右邊所有元素進行比較的時候,這種情況下可以用歸併排序,來解出一個數組中逆序數對,這個貌似是流行了十年多的經典應用了。


老規矩先說下什麼是逆序對,Int[] arr = {9,8,7},我們先定義一個方向,從左往右,任意取兩個數,只要左邊比右邊大,就算一個逆序對。arr的逆序對爲(9,8),(9,7),(8,7)。求解思路很簡單,就是在歸併的過程中,會不斷比較,每個數在比較過程中記錄下左大右小的兩個數就行了,不理解可以看下我的上一篇。ps:方向問題別想多了,有人問如果不看方向怎麼求,那就是求每個數比它小的個數然後求和。。

代碼和歸併排序差的不多,如下:

package class03;

/**
 * @Description 歸併排序非遞歸,求逆序數對
 * @Package: class03
 * @Author: Ray
 * @CreateTime: 2020/7/4 22:28
 * @E-mail: [email protected]
 */
public class Test9 {
	
	static int count = 0;//記錄逆序對個數
	
	public static int mergeSort(int[] arr) {
		if (arr == null || arr.length == 1) {   //空或一個數直接返回0
			return 0;
		}
		//初始跨度爲1
		int span = 1;
		//例如:span=1 -> 數組分割爲2個元素一組進行歸併,span=2 -> 數組分割爲4個元素一組來歸併 ……
		while (span < arr.length) {
			//for循環裏每2個span進行截取出left,mid,right,然後merge排序
			for (int i = 0; i < arr.length; i += 2 * span) {
				int left = i;
				int mid = left + span - 1;  //2個跨度進行歸併,mid爲1個跨度的位置
				/*處理邊界,當最後一個mid定位超出了長度,表示最後幾個數湊不到一個span,
				本次span長度=上次while裏span2倍,故也湊不齊上次while循環裏的2*span,
				湊不齊上次循環2個span的幾個數在上次while裏一定歸併排序好了,
				所以可以直接break
				 */
				if (mid >= arr.length - 1) {
					break;
				}
				int right = left + 2 * span - 1;
				if (right >= arr.length - 1) {  //處理邊界,右邊界超出長度則直接爲最後一位
					right = arr.length - 1;
				}
				merge(arr, left, mid, right);   //每個範圍歸併排序
			}
			//span = span << 1;   //優化一下可以用位運算左移,等同於乘2
			span = span * 2;    //跨度翻倍,翻倍變爲2,4,8,16……從而以log(N)次循環整個數組長度
		}
		return count;
	}
	
	//對排完序的left ~ mid, mid+1 ~ right範圍進行歸併排序
	public static int merge(int[] arr, int left, int mid, int right) {
		int[] tmp = new int[right - left + 1];  //臨時數組
		int l = left;       //左指針初始位置
		int m = mid + 1;    //右指針初始位置
		int i = 0;          //tmp下標
		while (l <= mid && m <= right) {    //任意左或右指針到邊界則退出
			if (arr[l] > arr[m]) {
				// 只有當左大右小,產生逆序對,因爲有序,l開始到mid每個數都比arr[m]大,
				// 逆序數就爲(arr[l]arr[m]),(arr[l+1]arr[m])……(arr[mid]arr[m])
				count = count + (mid - l + 1);    //逆序對個數累加
				for (int j = l; j <= mid; j++) {
					System.out.println("(" + arr[j] + arr[m] + ")");
				}
				tmp[i] = arr[m];
				m += 1;
			} else {
				tmp[i] = arr[l];
				l += 1;
			}
			i += 1;
		}
		while (l <= mid) {      //如果左指針先循環完,剩下的數補到tmp上
			tmp[i] = arr[l];
			l += 1;
			i += 1;
		}
		while (m <= right) {    //如果右指針先循環完,剩下的數補到tmp上
			tmp[i] = arr[m];
			m += 1;
			i += 1;
		}
		for (int j = 0; j < tmp.length; j++) {  //tmp還原到原數組
			arr[left + j] = tmp[j];
		}
		return count;
	}
	
	public static void main(String[] args) {
		int[] arr = {9,8,7,6};
		int count = mergeSort(arr);
		System.out.println("共產生" + count + "個逆序對");
		
	}
	
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章