题目描述:
在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。求一个数组 的小和
题目举例:
[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 lesson01;
public class demo02_SmallSum {
public static void main(String[] args) {
int arr[] = { 1, 3, 4, 2, 5 };
System.out.println("小和为:" + smallSum(arr));
// 最终归并排序的结果,本题目中没用
/*
* for(int i = 0; i < arr.length; i++) System.out.println(arr[i]);
*/
}
public static int smallSum(int[] arr) {
if (arr == null || arr.length == 1)
return 0;
return mergeSort(arr, 0, arr.length - 1);
}
public static int mergeSort(int[] arr, int l, int r) {
if (l == r)
return 0;
int mid = l + ((r - l) >> 1); // 向右移动一位,等于乘0.5
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; // 指针p1,p2等于将一个数组,虚拟成二个数组了
int p2 = mid + 1;
int sum = 0;
while (p1 <= mid && p2 <= r) {
sum += arr[p1] < arr[p2] ? (r - p2 + 1) * arr[p1] : 0; // 统计较小的数
help[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++]; // 哪个数组中的数小,哪个被放在help[]里边,然后i p*指针各自++
}
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;
}
}