归并排序之求小和

小和问题

在一个数组中，每一个数把左边比自己小的数累加起来，最后把所有的和累加起来叫做这个数组的小和。也就是说，在数组中看这个数右边有几个数比自己大，如果有一个数比自己大，就是1乘以自身，如果有两个数比自己大，就是2乘以自身，最后把每个数得到的乘积相加。

举例

现有数组【4,2,5,1,3】

对于4，右边有一个数5比自己大，就是 1*4

对于2，右边有两个数（5,3）比自己大，就是 2*2

对于5，右边有0个数比自己大，就是 0*5

对于1，右边有一个数3比自己大，就是 1*1

对于3，右边有0个数比自己大，就是 0*3

则小和为：1*4+2*2+0*5+1*1+0*3 = 9

可以通过归并排序算法解决上面的问题

代码

package sequence;

/**
 * 求小和
 */
public class SmallSum {

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

    }

    public 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);

    }

    public static int merge(int[] arr,int L,int mid,int R){
        int[] temp = new int[R-L+1];
        int i = 0;
        int p1 = L;
        int p2 = mid+1;
        int 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++];
        }

        System.arraycopy(temp,0,arr,L,temp.length);

        return res;

    }

    public static void main(String[] args) {
        System.out.println("小和："+smallSum(new int[]{4,2,5,1,3}));
    }


}

 

转载于:https://www.cnblogs.com/naloy/p/9943874.html