小和问题:
在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。求一个数组 的小和。
例子:
[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);
// }
}
}