一、小和问题
1、题目描述:
在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。求一个给定数组的小和。
2、例子:
数组为:[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
3、思路:
找每一个数右边比当前数大的个数,(个数 * 当前数) 的累加和就是结果。
这咋和归并排序联系上的呢,仔细想想,在左组和右组merge的时候,会比较数的大小,这时就可以在右组找到比左组当前数大的个数。
4、详细的参考代码:
/**
* 小和问题:在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。要求时间复杂度O(N*logN)
*
* @author Java和算法学习:周一
*/
public class SmallSum {
public static int smallSum(int[] arr) {
if (arr == null || arr.length < 2) {
return 0;
}
return process(arr, 0, arr.length - 1);
}
private 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[] help = new int[r - l + 1];
int i = 0;
int pL = l;
int pR = mid + 1;
int res = 0;
while (pL <= mid && pR <= r) {
// 当左组的数小于右组的数时, 当前右组的个数*当前数 的累加和 即是小和的结果
// 仔细和归并排序比较,发现就多了此处的代码。唯一的区别是,
// 等于的时候拷贝右组的,因为要在右组中找出比左组大的个数,肯定不能先拷贝左组的,不然咋找出个数
res += arr[pL] < arr[pR] ? (r - pR + 1) * arr[pL] : 0;