逆向思维,看左边小的数,其实也可以看右边大的数,例如
1,右边比它大的有四个,就会产生4*1=4的小何
3,右边比它大的有两个,就会产生3*2=6的小何
4,右边比它大的有一个,就会产生4*1=1的小何
2,右边比它大的有一个,就会产生2*1=2的小何
加起来也是16
运用归并排序的思想
把一个数组拆分成左右两部分(有序的两部分)然后求小和
左右两边各有一个指针,比1大的右边有两个,这个两个不是遍历得出来的,因为比右边指针位置小,右边又是有序的,所以比指针之后所有的数字都小,所以小和数加2
然后将指针左边指针指到下一个数,并且将1放到新开数组中,排序
上面的为特殊情况,这样的情况下必须先把右边的放到新开数组中排序,因为算小和时,左边和右边相等,不知道右边有几个数比左指针大,只有把右边相等的数都放到新开数组中,才能够知道右边有几个比左指针的数大。这一点是和经典的归并算法不同的。
public static int smallSum(int[] arr) {
if(arr.length<2 || arr == null) {
return -1;
}
return process(arr,0,arr.length-1);
}
public static int process(int[] arr,int L,int R) {
if(L==R) {
return 0;
}
int mid = L + ((R-L)>>1);
return process(arr,L,mid)+ process(arr,mid+1,R)+ merge(arr,L,mid,R);
//返回的是左边的小和加上右边的小和,然后他们归并时产生的小和
}
public static int merge(int[] arr,int L,int M,int R) {
int[] help = new int[R-L+1];
int p1 =L;
int p2 = M+1;
int sum = 0;//小和
int i = 0;
while(p1<=M&&p2<=R) {
if(arr[p1]<arr[p2]) {
help[i++] = arr[p1];
sum = sum+(R-p2+1)*arr[p1++];//只有在左边指针比右边小的时候才产生小和
}
else if(arr[p1]==arr[p2]) {
help[i++] = arr[p2++];
}
else if(arr[p1]>arr[p2]) {
help[i++] = arr[p2++];
}
}
while(p1<=M) {
help[i++] = arr[p1++];
}
while(p2<=R) {
help[i++] = arr[p2++];
}
for(int j=0;j<help.length;j++) {
arr[L+j] = help[j];//拷贝排序数组
}
return sum;//返回归并时候产生的小和
}