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));
		
	}
	
}

 

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