数组单调和

题目描述

现定义数组单调和为所有元素i的f(i)值之和。这里的f(i)函数定义为元素i左边(不包括其自身)小于等于它的数字之和。请设计一个高效算法，计算数组的单调和。

给定一个数组A同时给定数组的大小n，请返回数组的单调和。保证数组大小小于等于500，同时保证单调和不会超过int范围。

测试样例：
[1,3,5,2,4,6],6
返回：27

解法

1.蛮力法
直接搜索求和，复习度O(n^2),实现代码如下：

import java.util.*;

public class MonoSum {
    public int calcMonoSum(int[] A, int n) {
        // write code here
        int sum = 0;
        for(int i = 1;i < n;i++){
            for(int j = i - 1;j >= 0;j--){
                if(A[j] <= A[i]){
                    sum = sum + A[j];
                }
            }
        }
        return sum;
    }
}

2.归并排序
利用归并排序特性，当左边子数组的数归并时，将其乘以右边子数组的元素数量，算入总数中，以此求出单调和，
举例：

{1，3，4，6}
{1，3}归并时区分出{1}，{3}，算出单调和为1
{4，6}同理，算出4
{1，3}和{4，6}，算出1 * 2 + 3 * 2 = 8
单调和为 sum = 1 + 4 + 8 =13

蛮力法为：1 + 1 + 3 + 1 + 3 + 4 =13
答案相同

此方法复杂度为O(nlogn),具体实现代码如下：

import java.util.*;

public class MonoSum {
    private int dp = 0;
    public int calcMonoSum(int[] A, int n) {
        // write code here
        mergesort(A,0,n - 1);
        return dp;
    }
    public void mergesort(int[] A,int start,int end){
    	if(end <= start){
            return ;
        }
        int middle = start + (end - start)/2;
        mergesort(A,start,middle);
        mergesort(A,middle + 1,end);
        merge(A,start,middle,end);
    }
    public void merge(int[] A,int start,int middle,int end){
        int[] temp = new int[A.length];
        for(int i = start;i <= end;i++){
            temp[i] = A[i];
        }
        int i1 = start;
        int i2 = middle + 1;
        for(int i = start;i <= end;i++){
            if(i1 > middle){
                A[i] = temp[i2];
                i2++;
            }
            else if(i2 > end){
                A[i] = temp[i1];
                i1++;
            }
            else if(temp[i1] > temp[i2]){
                A[i] = temp[i2];
                i2++;
            }
            else{
            	dp = dp + temp[i1] * (end - i2 + 1);
                A[i] = temp[i1];
                i1++;
            }
        }
    }
}