在一个数组中,每一个元素左边比当前元素值小的元素值累加起来,叫做这个数组的小和
例如:[2,3,4,1,5]
2左边比2小的元素:无
3左边比3小的元素:2
4左边比4小的元素:2,3
1左边比1小的元素:无
5左边比5小的元素:2,3,4,1
小和small_sum = 2 + 2 + 3 + 2 + 3 + 4 + 1 = 17
思路:使用归并算法的思想进行求解
public static int smallSum(int[] arr){
if(arr == null || arr.length < 2){
return 0;
}
return mergrSort(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) >> 2; //找到中点
return mergeSort(arr, L, mid) + mergeSort(arr, mid + 1, R) + merge(arr, L, mid, R);
}
public static merge(int[] arr, int L, int mid, int R){
int[] help = new int[R - L + 1];
int p1 = L;
int p2 = mid + 1;
int res = 0;
int i = 0;
while(p1 <= m && p2 <= R){
res += arr[p1] < arr[p2] : (r - p2 + L) * arr[p1] : 0;
//将排序后的数放入辅助数组
help[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
}
while(p1 <= L){//若p2耗尽则将p1中剩余的数字填入help中
help[i++] = arr[p1++];
}
while(p2 <= R){//若p1耗尽则将p2中剩余的数字填入help中
help[i++] = arr[p2++];
}
for(int i = 0; i < help.length; i++){//将辅助数组中的数拷贝到原数组中
arr[L + i] = help[i];
}
return res;
}