小和问题(递归法)

求一个数组的小和。

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

【算法分析】

比如想要求解数组[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

public class SmallSum {
    
    public static void main(String[] args) {
        int[] arr = {3, 5, 6, 9, 6, 2, 4, 2};
        int sum = 0;
        for (int i = 1; i < arr.length; i++) {
            for (int j = 0; j < i; j++) {
                if (arr[j] < arr[i]) {
                    sum += arr[j];
                }
            }
        }
        System.out.println(sum);

        System.out.println(process(arr, 0, arr.length - 1));
    }

    /**
     * 递归过程
     *
     * @param arr 原数组
     * @param L   左边界
     * @param R   右边界
     * @return
     */
    public static int process(int[] arr, int L, int R) {
        if (L == R) {
            return 0;
        }
        int mid = L + ((R - L) >> 1);
        return process(arr, L, mid) + process(arr, mid + 1, R) + merge(arr, L, mid, R);
    }

    private static int merge(int[] arr, int L, int mid, int R) {
        //辅助数组
        int[] helpArr = new int[R - L + 1];
        int helpIndex = 0;
        int sum = 0;
        //左边界起始位置
        int leftIndex = L;
        //右边界起始位置
        int rightIndex = mid + 1;
        //左右两边对比,如果左和右都不越界
        while (leftIndex <= mid && rightIndex <= R) {
            //如果左边的数比右边最小的数小,那么sum就+=(R - rightIndex + 1) * arr[leftIndex],否则就+0
            sum += arr[leftIndex] < arr[rightIndex] ? (R - rightIndex + 1) * arr[leftIndex] : 0;
            helpArr[helpIndex++] = arr[leftIndex] < arr[rightIndex] ? arr[leftIndex++] : arr[rightIndex++];
        }
        while (leftIndex <= mid) {
            helpArr[helpIndex++] = arr[leftIndex++];
        }
        while (rightIndex <= R) {
            helpArr[helpIndex++] = arr[rightIndex++];
        }
        for (int i = 0; i < helpArr.length; i++) {
            arr[L + i] = helpArr[i];
        }
        return sum;

    }
}

第一种写法是 O(N2)
第二种写法是O(NlogN)
第二种思路:归并排序思路
每次排好一个步长,如果右部分比左部分的值大,则sum+=(R-rightIndex+1)*helpArr[leftIndex],否则加0.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值