求一个数组的小和。
什么是小和?
在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。
【算法分析】
比如想要求解数组[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
public class SmallSum {
public static void main(String[] args) {
int[] arr = {3, 5, 6, 9, 6, 2, 4, 2};
int sum = 0;
for (int i = 1; i < arr.length; i++) {
for (int j = 0; j < i; j++) {
if (arr[j] < arr[i]) {
sum += arr[j];
}
}
}
System.out.println(sum);
System.out.println(process(arr, 0, arr.length - 1));
}
/**
* 递归过程
*
* @param arr 原数组
* @param L 左边界
* @param R 右边界
* @return
*/
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);
}
private static int merge(int[] arr, int L, int mid, int R) {
//辅助数组
int[] helpArr = new int[R - L + 1];
int helpIndex = 0;
int sum = 0;
//左边界起始位置
int leftIndex = L;
//右边界起始位置
int rightIndex = mid + 1;
//左右两边对比,如果左和右都不越界
while (leftIndex <= mid && rightIndex <= R) {
//如果左边的数比右边最小的数小,那么sum就+=(R - rightIndex + 1) * arr[leftIndex],否则就+0
sum += arr[leftIndex] < arr[rightIndex] ? (R - rightIndex + 1) * arr[leftIndex] : 0;
helpArr[helpIndex++] = arr[leftIndex] < arr[rightIndex] ? arr[leftIndex++] : arr[rightIndex++];
}
while (leftIndex <= mid) {
helpArr[helpIndex++] = arr[leftIndex++];
}
while (rightIndex <= R) {
helpArr[helpIndex++] = arr[rightIndex++];
}
for (int i = 0; i < helpArr.length; i++) {
arr[L + i] = helpArr[i];
}
return sum;
}
}
第一种写法是 O(N2)
第二种写法是O(NlogN)
第二种思路:归并排序思路
每次排好一个步长,如果右部分比左部分的值大,则sum+=(R-rightIndex+1)*helpArr[leftIndex],否则加0.