Java_Basic_smallsum_review

【学习来自 -- 左老师】

【java 基础】： 小和问题

 

【代码如下】

public class SmallSum_01 {

	public static void main(String[] args) {
		int[] arr1 = { 1, 4, 3, 5};
		smallSum(arr1);
	}
	
	public static int smallSum(int[] arr) {
		// TODO Auto-generated method stub
		if (arr == null || arr.length < 2) {
			return 0;
		}
		return mergeSort(arr, 0, arr.length-1);
	}
	
	// 返回L~R 返回多少小和
	public static int mergeSort(int[] arr, int L, int R) {
		if ( L == R) {
			return 0 ;
		}
		// 等同于 （L+R）/2, 这样可能溢出
		int mid = L + ((R - L) >>1);
		// 左侧排序中产生的小和，右侧排序产生的小和，整体合并中产生的小和
		return mergeSort(arr,L, mid) + mergeSort(arr, mid+1,R) + merge(arr, L, mid, R);
		
	}
	
	public static int merge(int[] arr, int L, int mid, int R) {
		int[] help = new int[R - L +1];
		int i = 0;
		int p1 = L;
		int p2 = mid+1;
//		初始化结果的值 res
		int res = 0;
		while ( p1 <= mid && p2 <= R) {
			// 排序过程中产生小和，如果左侧数比右侧小，那么这个数*右侧的个数，注意下标
			res += arr[p1] < arr[p2] ? (R - L + 1) * arr[p1] : 0;
			help[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
		}
		while (p1 <= mid) {
			help[i++] = arr[p1++];
		}
		while (p2 <= R) {
			help[i++] = arr[p2++];
		}
		for ( i=0; i<help.length; i++) {
			arr[L+i] = help[i];
		}
		return res;		
	}

}