归并排序中的小和问题

小和定义：如果一个数组int[] arr，对每个元素arr[i]求 其左边比他小的元素的累加和sum[i]，这些累加和的总和，就叫小和。
也就是小和 s = sum[0] + sum[1] + … + sum[i] + … + sum[n-1]

其实求小和的问题，只需要在归并算法基础上，增加求和的逻辑，就🆗了

直接上代码

    public static int smallSum(int[] arr) {
        if (arr == null || arr.length < 2) {
            return 0;
        }

        return process(arr, 0, arr.length - 1);
    }

    /**
     * 求数组 从下标 L到R 上即[L,R]的小和
     * @param arr
     * @param L
     * @param R
     * @return
     */
    public static int process(int[] arr, int L, int R) {
        //用大腿想一想，如果只有一个元素，那小和肯定是0啊
        if (L == R) {
            return 0;
        }
        //归并算法下标中间值。
        int M = L + ((R - L) >> 1);
        //把[L,R]数组，从中间M分开左右两边，左侧的小和+右侧的小和+合并求和，就是最终的结果
        return process(arr, L, M)
                + process(arr, M + 1, R)
                + merge(arr, L, M, R);
    }

    /**
     * 真正计算的小和，其实是在merge中计算的
     * @param arr
     * @param L
     * @param M
     * @param R
     * @return
     */
    public static int merge(int[] arr, int L, int M, int R) {
        int[] help = new int[R - L + 1];
        int p1 = L;
        int p2 = M + 1;
        int i = 0;
        int res = 0;
        while (p1 <= M && p2 <= R){
            //如果p1指针对用的元素，小于p2指针的元素，说明这个时候需要把p1指针的元素添加到辅助数组help中
            //这个时候，就会知道，右侧的分数组即[M+1,R]上的这个数组，比arr[p1]大的元素有 R-(p2-1)个
            res += arr[p1] < arr[p2] ? (R - p2 + 1) * arr[p1] : 0;
            help[i++] = arr[p1] < arr[p2] ? arr[p1++]:arr[p2++];
        }
        while (p1 <= M){
            help[i++] = arr[p1++];
        }
        while (p2 <= R){
            help[i++] = arr[p2++];
        }
        for (int j = 0; j < help.length; j++) {
            arr[L + j] = help[j];
        }
        return res;
    }

验证：

    public static void main(String[] args) {
        int[] arr = {4, 2, 5, 7, 1, 9, 0};
        CommonUtils.printArray(arr);
        System.out.println("小和 = " + smallSum(arr));
        CommonUtils.printArray(arr);
    }