在一個數組中,每一個元素左邊比當前元素值小的元素值累加起來,叫做這個數組的小和
例如:[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;
}