小和问题

小和问题

描述

在一个数组中，每一个数左边比当前数小的数累加起来，叫做这个数组的小和。求一个数组的小和。

例子

[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。而且这样是不会加重复的，因为合并完后，他们就属于同一边了，同一边的数是不会在去计算小和的。