【笔试算法】小和问题

小和问题：

在一个数组中，每一个数左边比当前数小的数累加起来，叫做这个数组的小和。求一个数组 的小和。

例子:
[1,3,4,2,5]

1左边比1小的数，没有;

3左边比3小的数，1;

4左边比4小的数，1、3;

2左边比2小的数，1;

5左边比5小的数，1、3、4、2;

所以小和为1+1+3+1+1+3+4+2=16

 

代码：

package cn.itcats.recursion;

import org.junit.Test;

/**
 * 使用归并排序完成小和问题
 * @author fatah
 *
 */
public class SmallSum {
	public int smallSum(int[] arr) {
		if(arr == null || arr.length < 2) {
			return 0;
		}
		return mergeSort(arr,0,arr.length-1);
	}

	public int mergeSort(int[] arr, int l, int r) {
		if(l == r) {
			return 0;
		}
		int mid = r + ((l - r) >> 1);
		return mergeSort(arr,l,mid)+
				mergeSort(arr, mid+1, r)+
				merge(arr, l, mid , r);
		
	}

	public 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 ;
		int sum = 0;
		while(p1 <= mid && p2 <= r) {
			sum += arr[p1] < arr[p2]?arr[p1] * (r - p2 + 1):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(int j =0 ;j < help.length ; j++) {
			arr[l+j] = help[j];
		}
		return sum;
	}
	
	@Test
	public void Test() {
		int [] arr = {1,2,3,4};
		System.out.println(smallSum(arr));;
//		for(int i : arr) {
//			System.out.println(i);
//		}
	}
}