左神算法 计算数组小和

给定数组,对于每个值来说,它左边的小于等于它的数之和称为小和,返回整个数组的小和之和。
这里换个思路,数组分一半,求出左边的某个值比右半数组多少个数小,这是数组间的小和累加,再求左半边数组和右半边数组的小和,相当于将问题的规模减小了。
归并排序的思想
下面的是代码:

public class Main {
    public static void main(String[] args) {
        int[] arr = { 3, 1, 2, 4, 6, 2, 7, 8, 1, 7,9,0,4,5,6,7,8,2};
        int[] tmp = new int[arr.length];
        System.out.println(getSmallSum(arr, 0, arr.length -1, tmp));
    }

    public static int getSmallSum(int[] arr, int start, int end, int[] tmp){
        if (start >= end){
            return 0;
        }
        int mid = (end-start)/2 + start;
        int res1 = getSmallSum(arr, start, mid, tmp);
        int res2 = getSmallSum(arr, mid +1 , end, tmp);
        int res3 = merge(arr, start, mid, end, tmp);
        return res1 + res2 + res3;
    }

    public static int merge(int[] arr, int start, int mid, int end, int[] tmp){
        int res = 0;
        int left = start, right = mid +1;
        int index = left;
        while (left <= mid && right <= end){
            if (arr[left] <= arr[right]){
                res += (end-right+1) * arr[left];
                tmp[index] = arr[left];
                ++left;
            }
            else {
                tmp[index] = arr[right];
                ++right;
            }
            index++;
        }
        while (left <= mid){
            tmp[index] = arr[left];
            ++index;
            ++left;
        }
        while (right <= end){
            tmp[index] = arr[right];
            ++index;
            ++right;
        }
        for (int i = start; i <= end; i++) {
            arr[i] = tmp[i];
        }
        return res;
    }


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值