计算数组的小和

【题目】

数组小和的定义如下：
例如，数组s=[1,3,5,2,4,6]，在s[0]的左边小于或等于s[0]的数的和为0，在s[1]的左边小于或等于s[1]的数的和为1，在s[2]的左边小于或等于s[2]的数的和为1+3=4，在s[3]的左边小于或等于s[3]的数的和为1，在s[4]的左边小于或等于s[4]的数的和为1+3+2=6，在s[5]的左边小于或等于s[5]的数的和为1+3+5+2+4=15，所以s的小和为0+1+4+1+6+15=27。

给定一个数组s，实现函数返回s的小和。

【代码】

public class demo {
    public static void main(String[] args) {
        int[] arr=new int[]{1,3,5,2,4,6};
        System.out.println(smallSum(arr));//27
        System.out.println(smallSum2(arr));//27

        int[] arr1=new int[]{7,6,5,3,1,6};
        System.out.println(smallSum(arr1));//15
        System.out.println(smallSum2(arr1));//15
    } 

    //计算数组的小和
    //方法1：时间复杂度O(N^2)
    public static int smallSum(int[] arr){
        if(arr==null||arr.length==0){
            return 0;
        }
        int[] dp=new int[arr.length];
        dp[0]=0;
        int sum1=0;
        for(int i=1;i<arr.length;i++){
            for(int j=i-1;j>=0;j--){
                if(arr[j]<=arr[i]){
                    dp[i]+=arr[j];
                }
            }
            sum1+=dp[i];
        }
        return sum1;
    }

    //方法2：时间复杂度O(NlogN)，额外空间复杂度O(N)
    public static int smallSum2(int[] arr){
        if(arr==null||arr.length==0){
            return 0;
        }
        return func(arr,0,arr.length-1);//归并排序
    }

    private static int func(int[] s, int l, int r) {
        if(l==r){
            return 0;
        }
        int mid=(l+r)/2;
        return func(s,l,mid)+func(s,mid+1,r)+merge(s,l,mid,r);
        //归并排序的过程会进行拆组再合并，即：拆左组，拆右组，合并左右组
    }

    private static int merge(int[] s, int left, int mid, int right) {
        int[] h=new int[right-left+1];
        int hi=0;//h的首位
        int i=left;//左组首位
        int j=mid+1;//右组首位
        int sum=0;
        while(i<=mid && j<=right){
            if(s[i]<=s[j]){
                sum+=s[i]*(right-j+1);//生成小和
                h[hi++]=s[i++];//左组动
            }
            else{
                h[hi++]=s[j++];//右组动
            }
        }
        for(;(j<right+1)||(i<mid+1);j++,i++){
            h[hi++]=i>mid?s[j]:s[i];//左右组合成h
        }
        for(int k=0;k!=h.length;k++){
            s[left++]=h[k];//合成s
        }
        return sum;
    }