题目描述
现定义数组单调和为所有元素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++;
}
}
}
}