小和问题
描述
在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。求一个数组的小和。
例子
[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
最简单的方法是用两层for循环,当前值比他之前的值大,就当之前的值累加,这样最后就可以计算出小和,时间复杂度为O(n^2),但是这样解效率太差,所以不推荐这种解法。还可以使用分治来解这道题,引用官方的话来说:分治,字面上的解释是“分而治之”,就是把一个复杂的问题分成两个或更多的相同或相似的子问题,再把子问题分成更小的子问题……直到最后子问题可以简单的直接求解,原问题的解即子问题的解的合并。
public class MinSum {
public static int minSum(int[] arr){
int result = minSum(arr,0,arr.length - 1);
return result;
}
private static int minSum(int[] arr, int L, int R) {
if(L == R){
return 0;
}
int mid = L + ((R - L) >> 1);
int left = minSum(arr, L, mid);
int right = minSum(arr,mid + 1,R);
int mergeSum = merge(arr,L,mid,R);
return left + right +mergeSum;
}
private static int merge(int[] arr, int L, int mid, int R) {
int help[] = new int[R - L +1];
int p1 = L;
int p2 = mid + 1;
int sum = 0;
int current = 0;
while(p1 <= mid && p2 <= R){
sum += arr[p1] < arr[p2] ? arr[p1] * (R - p2 + 1) : 0;
help[current++] = arr[p1] > arr[p2] ? arr[p2++] : arr[p1++];
}
while (p1 <= mid){
help[current++] = arr[p1++];
}
while (p2 <= R){
help[current++] = arr[p2++];
}
for(int i = 0;i < help.length;i++){
arr[L + i] = help[i];
}
return sum;
}
}
这个问题等同于归并排序,归并排序在合并时划分的左右两边都是有序的。在合并的时候,当两个指针指向的左边的数比右边的数小时,那么右边下标[p2-R]上的数都会比左边这个数大,小和结果就需要加上左边相对较小的那个数乘以R-p2。而且这样是不会加重复的,因为合并完后,他们就属于同一边了,同一边的数是不会在去计算小和的。