问题描述
在一个数组中, 每一个数左边比当前数小的数累加起来, 叫做这个数组的小和。 求一个数组
的小和。
例子:
[1,3,4,2,5
]
1左边比1小的数, 没有;
3左边比3小的数, 1;
4左边比4小的数, 1、 3;
2左边比2小的数, 1;
5左边比5小的数, 1、 3、 4、 2;
所以小和为1+1+3+1+1+3+4+2=16
解决思路
最简单的应该是一次遍历,但时间复杂度很高。转换一下思路可以变为右边比它大的数字有几个自身便加几次。所以数字本身便会和其右边数依次比较,此时时间复杂度能是o(n^2),考虑归并排序的思想如果比较时其右边是有序的(相当于把之前的比较结果进行了存储),那么便可以减少比较的次数,时间复杂度降到o(n*logn)。
public static int sum(int []a,int l,int r) {
if(l==r) {
return 0;
}
int mid=l+((r-l)>>1);
return sum(a,l,mid)+sum(a,mid+1,r)+merge(a,l,r);
}
private static int merge(int[]a,int l, int r) {
// TODO Auto-generated method stub
int mid=l+((r-l)>>1);
int po=l;
int p=mid+1;
int []help=new int[r-l+1];
int i=0;
int ans=0;
while(po<=mid&&p<=r) {
if(a[po]<a[p]) {
ans=ans+a[po]*(r-p+1);
help[i++]=a[po++];
}else {
help[i++]=a[p++];
}
}
while(po<=mid) {
help[i++]=a[po++];
}
while(p<=r) {
help[i++]=a[p++];
}
i=0;
while(i<=r-l) {
a[l++]=help[i++];
}
return ans;
}