1. 问题描述
数组小和的定义如下:
例如,数组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的小和。
input: [1, 3, 5, 2, 4, 6]
output: 27
2. 解法
最简单的就是遍历,时间复杂度为 O ( N 2 ) O(N^2) O(N2)。这里我们如果想要减少复杂度,比如达到 O ( N ∗ l o g 2 N ) O(N * log_{2}N) O(N∗log2N)。我们要转变思想把当前数前面比她小的和转变为当前数要被加几次(就是跟后面数做比较,后面数比当前数大,就要加一次),可以先分成两半,求每一半的小和,然后在将其合并,在更大的范围内求小和,这个正符合归并排序,只是在归并排序的基础上简单的进行修改。每次比较,我们需要改进的是在比较是若后面数比当前数大,res就加当前数,只需加一句:
res += vec[p1] < vec[p2] ? (r - p2 + 1) * vec[p1] : 0;
下面我们还是看代码:
C++解法:
#include <assert.h>
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <vector>
template<typename T>
bool compare(T a, T b) {
return a < b;
}
template<typename T>
void rightMethod(std::vector<T>& vec) {
sort(vec.begin(), vec.end(), compare<T>);
}
template<typename T>
int mergeVector(std::vector<T>& vec, int start, int middle, int end) {
assert(middle >= start);
assert(end >= middle);
std::vector<T> help (end - start + 1, -1);
int p1 = start;
int p2 = middle + 1;
int index = 0;
int res = 0;
while ((p1 <= middle)&& (p2 <= end)) {
res += vec[p1] < vec[p2] ? ((end - p2 + 1) * vec[p1]) : 0;
help[index++] = vec[p1] < vec[p2] ? vec[p1++] : vec[p2++];
}
while (p1 <= middle) {
help[index++] = vec[p1++];
}
while (p2 <= end) {
help[index++] = vec[p2++];
}
for (int i = 0; i < help.size(); ++i) {
vec[i + start] = help[i];
}
std::vector<T> ().swap(help);
return res;
}
template<typename T>
int partitionMergeSort(std::vector<T>& vec, int start, int end) {
if (start == end) {
return 0;
}
int middle = ((end - start) >> 1) + start;
return partitionMergeSort(vec, start, middle) +
partitionMergeSort(vec, middle + 1, end) +
mergeVector(vec, start, middle, end);
}
template<typename T>
int smallSum(std::vector<T>& vec) {
int length = vec.size();
if (length < 2) {
return 0;
}
int start = 0;
int end = length - 1;
int res = partitionMergeSort(vec, start, end);
return res;
}
template<typename T>
void inputVector(std::vector<T>& vec) {
T data;
std::cin >> data;
vec.push_back(data);
while (std::cin.get() != '\n') {
std::cin >> data;
vec.push_back(data);
}
}
int main() {
std::vector<int> vec;
inputVector(vec);
int res = smallSum(vec);
std::cout << "small sum: " << res << std::endl;
return 0;
}
3. 后记
刚开始按照归并排序进行修改,一直没有得到正确的结果,于是想起了在codeblocks进行debug, 果然解决了问题的症结所在。