2020-11-4 计算数组的小和

题目描述

数组小和的定义如下：

例如，数组s = [1, 3, 5, 2, 4, 6]，在s[0]的左边小于或等于s[0]的数的和为0；在s[1]的左边小于或等于s[1]的数的和为1；在s[2]的左边小于或等于s[2]的数的和为1+3=4；在s[3]的左边小于或等于s[3]的数的和为1；

在s[4]的左边小于或等于s[4]的数的和为1+3+2=6；在s[5]的左边小于或等于s[5]的数的和为1+3+5+2+4=15。所以s的小和为0+1+4+1+6+15=27

给定一个数组s，实现函数返回s的小和

[要求]

时间复杂度为O(nlogn)，空间复杂度为O(n)

输入描述:

第一行有一个整数N。表示数组长度
接下来一行N个整数表示数组内的数

输出描述:

一个整数表示答案

示例1

输入

复制

6
1 3 5 2 4 6

输出

复制

27

备注:

1⩽N⩽10^5

−100⩽arri⩽100

#include <bits/stdc++.h>
using namespace std;


long merge(vector<int>& arr, int left, int mid, int right) {
    vector<int> help(right - left + 1 , 0);
    long res = 0;
    int index = 0;
    int p1 = left, p2 = mid + 1;
    while(p1 <= mid && p2 <= right) {
        res += arr[p1] <= arr[p2] ? arr[p1] * (right - p2 + 1) : 0;
        help[index ++] = arr[p1] <= arr[p2] ? arr[p1 ++] : arr[p2 ++];
    }
    while(p1 <= mid) help[index ++] = arr[p1 ++];
    while(p2 <= right) help[index ++] = arr[p2 ++];
    for(int i = 0; i < help.size(); i ++) arr[left + i] = help[i];
    return res;
}



long mergeSort(vector<int>& nums, int l, int r) {
    if(l >= r) return 0;
    int mid = l + ((r - l) >> 1);
    return mergeSort(nums, l, mid) + mergeSort(nums, mid + 1, r) + merge(nums, l, mid, r);
}
 

int main() {
    int n;
    while(cin >> n) {
        vector<int> nums(n, 0);
        for(int i = 0; i < n; i ++) {
            cin >> nums[i];
        }
        cout << mergeSort(nums, 0, n - 1) << endl;
    }
    return 0;

}