归并排序--小和问题

问题描述

在一个数组中， 每一个数左边比当前数小的数累加起来， 叫做这个数组的小和。 求一个数组的小和。
样例

[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

思路：归并排序

#include <iostream>
using namespace std;

int merge(int array[], int L, int mid, int R){
    int help[R - L + 1];
    int i = 0;
    int p1 = L;
    int p2 = mid + 1;
    int res = 0;
    while (p1 <= mid && p2 <= R){
        res += array[p1] < array[p2] ? (R - p2 + 1)*array[p1] : 0;
        help[i++] = array[p1] < array[p2] ? array[p1++] : array[p2++];
    }
    while (p1 <= mid){
        help[i++] = array[p1++];
    }
    while (p2 <= R){
        help[i++] = array[p2++];
    }
    for (i = 0; i <= R - L; i++){
        array[L+i] = help[i];
    }
    return res;
}

int mergeSort(int array[], int L, int R){
    if(L == R)
        return 0;
    //int mid = L + ((R - L)>> 1);
    int mid = (L + R) / 2;
    return mergeSort(array, L, mid)
            + mergeSort(array, mid+1, R)
            + merge(array, L, mid, R);
}

int SmallSum(int array[], int length){
    if (array ==NULL || length < 2)
        return 0;
    return mergeSort(array, 0, length - 1);
}

int main(){
    int array[101] = {4,2,5,1,7,3,6};
    int length = 7;
    int s = SmallSum(array, length);
    cout << s <<endl;
    return 0;
}