java歸併排序的應用--求小和

歸併排序大家都會寫了吧,那你知道什麼場景或什麼題能派上用場嗎?當涉及到數組元素中,每個元素要和左邊所有,或右邊所有元素進行比較的時候,這種情況下可以用歸併排序,在排序的過程中,把事兒幹了

不理解的話,看下這個圖就清楚了,我們現在只看3的心路歷程,在merge的過程中,3第一次和1merge比較,然後進行下一個範圍的merge,下下個……,每個新的範圍所有數都會和3比較,比較的過程中可以實現特有的邏輯,當範圍不斷擴大,直到數組末尾的時候,3所有右邊的數就完成了和3的比較過程。

現在有這麼個題,求一個int數組的小和,就可以用這個思路來解。


什麼是小和,int[] arr = {1, 2, 3, 2},小和就是每個元素拿出來,所有左邊比它小的數求和,再進行累加,比如arr裏1左邊沒數,sum1 = 0,第二個數sum2=1,下面sum3=1+2,sum4=1,小和=sum1+sum2+sum3+sum4 = 5 

這個題的思路有個轉變, {1,2,3,4}小和的值裏面有幾個1?右邊比1大的數都會產生1,共有3個,有幾個2呢,右邊比2大的數會產生2,共有2個,小和計算可以用這個思路轉化爲每個數右邊有幾個比它大,那麼就產生右邊那麼多個它

爲了簡單起見,用非遞歸的實現,邏輯直接和就是上圖,大部分代碼和歸併排序一致,只是merge方法裏稍有變化。


 

package class03;

/**
 * @Description 歸併排序非遞歸,求小和
 * @Package: class03
 * @Author: Ray
 * @CreateTime: 2020/7/4 15:28
 * @E-mail: [email protected]
 */
public class Test8 {
	
	public static int mergeSort(int[] arr) {
		if (arr == null || arr.length == 1) {   //空或一個數直接返回0
			return 0;
		}
		//初始跨度爲1
		int span = 1;
		int num = 0;
		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;
				}
				num += merge(arr, left, mid, right);   //每個範圍歸併排序,計算小和進行累加
			}
			span = span * 2;    //跨度翻倍,翻倍變爲2,4,8,16……從而以log(N)次循環整個數組長度
		}
		return num;
	}
	
	//對排完序的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下標
		int num = 0;        //小和
		while (l <= mid && m <= right) {    //任意左或右指針到邊界則退出
			if (arr[l] < arr[m]) {  //只有當左小右大,才進行求和
            //因爲是有序的,右邊m開始每個數都比l大,就產生(right-m+1)個arr[l]
				num += (right - m + 1) * arr[l];    
				tmp[i] = arr[l];
				l += 1;
			} else {
				tmp[i] = arr[m];
				m += 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 num;
	}
	
	public static void main(String[] args) {
		int[] arr = {1,2,3,2};
		System.out.println(mergeSort(arr));
		
	}
	
}

 

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