【题目】数组小和的定义如下:
例如,数组s=[1,3,5,2,4,6],在s[0]的左边小于或等于s[0]的数的和为0,在s[1]的左边小于或等于s[1]的数的和1,在s[2]左边小于等于s[2]的数的和为1+3=4,在s[3]的左边小于或等于s[3]的数的和为1,在s[4]的左边小于或等于s[4]的数的和为1+3+2=6,在s[5]的左边小于或等于s[5]的数的和1+3+5+2+4=15,所以s的小和为0+1+4+1+6+15=27.
给定一个数组s,实现函数返回s的小和。
public class SmallSum {
public static int mergeSort(int[] arr) {
if (arr == null || arr.length < 2) {
return 0;
}
return sortProcess(arr, 0, arr.length - 1);
}
public static int sortProcess(int[] arr, int l, int r) {
if (l == r) {
return 0;
}
int mid = l + ((r - l) >> 1);
return sortProcess(arr, l, mid) + sortProcess(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 pi = l;
int pj = mid + 1;
int smallSum = 0;
while (pi <= mid && pj <= r) {
smallSum += arr[pi] < arr[pj] ? arr[pi] * (r - pj + 1) : 0;
help[i++] = arr[pi] < arr[pj] ? arr[pi++] : arr[pj++];
}
while (pi <= mid) {
help[i++] = arr[pi++];
}
while (pj <= r) {
help[i++] = arr[pj++];
}
for (i = 0; i < help.length; i++) {
arr[l + i] = help[i];
}
return smallSum;
}
}