小和问题【归并排序】

题目描述

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

例子:

输入:[1, 3, 4, 2, 5]
输出:16

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

思路

实现的基本原理还是[归并排序]

  1. 将当前序列分为两个子序列,分别求其小和
  2. 对a划分得到的两个子序列进行merge操作,得到合并过程产生的小和,再加上a得到的两个子序列的小和之和
  3. 递归地执行1和2
private static int smallSum(int[] arr) {
    if (arr == null || arr.length < 2) {
        return 0;
    }
    return mergeSort(arr, 0, arr.length - 1);
}

private static int mergeSort(int[] arr, int l, int r) {
    if (l == r) {
        return 0;
    }
    int mid = l + ((r - l) >> 1);
    // 返回值为左右的小和加合并的小和
    return mergeSort(arr, l, mid) + mergeSort(arr, mid + 1, r) +
            merge(arr, l, mid, r);
}

private static int merge(int[] arr, int l, int mid, int r) {
    int[] temp = new int[r - l + 1];
    int i = 0, p1 = l, p2 = mid + 1, res = 0;
    while (p1 <= mid && p2 <= r) {
        // 此处处理累加的和
        res += arr[p1] < arr[p2] ? (r - p2 + 1) * arr[p1] : 0;
        temp[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
    }

    while (p1 <= mid) temp[i++] = arr[p1++];

    while (p2 <= r) temp[i++] = arr[p2++];

    for (i = 0; i < temp.length; i++) {
        arr[l + i] = temp[i];
    }
    return res;
}

public static void main(String[] args) {
    int[] arr = {1, 3, 4, 2, 5};
    int res = smallSum(arr);
    System.out.println(res);
}
发布了88 篇原创文章 · 获赞 13 · 访问量 2万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 精致技术 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览