介绍
在一个数组中,每个数左边比当前数小的累加起来,叫做这个数组的小和。求一个数组的小和。
思想
这道题是归并排序(点击链接查看归并排序有关内容)的典型应用,不难发现,在归并排序的过程中,我们将数组分为左右两个部分,分别进行排序,最后通过外排的方式,使得左右两个部分整体有序
我们可以在外排这个过程中求出小和:
即当左半部分的一个数小于右半部分的一个数时,那么一定构成小和,并且由于右半部分是从小到大有序的,因此右半部分这个数之后的所有数都与左半部分当前这个数构成小和
完整代码
#include <bits/stdc++.h>
using namespace std;
int merge(int arr[], int L, int mid, int R){
int result = 0;
int *helpArr = new int [R-L+1];
int i = 0;
int p1 = L;
int p2 = mid+1;
while(p1 <= mid && p2 <= R){
result += arr[p1] < arr[p2] ? (R-p2+1)*arr[p1] : 0;
helpArr[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
}
while(p1 <= mid){
helpArr[i++] = arr[p1++];
}
while(p2 <= R){
helpArr[i++] = arr[p2++];
}
for(i = 0; i < R-L+1; i++){
arr[L+i] = helpArr[i];
}
return result;
}
int sortProcess(int arr[], int L, int R){
if(L == R)
return 0;
int mid = L + ((R - L) >> 1);
/*sortProcess(arr, L, mid);
sortProcess(arr, mid+1, R);*/
return sortProcess(arr, L, mid) + sortProcess(arr, mid+1, R) + merge(arr, L, mid, R);
}
int mergeSort(int arr[], int size){
return sortProcess(arr, 0, size-1);
}
int insertSort(int arr[], int size){
int result = 0;
for(int i = 0; i < size; i++){
for(int j = i-1; j >= 0 && arr[j] < arr[j+1]; j--){
int temp = arr[j];
result += temp;
arr[j] =arr[j+1];
arr[j+1] = temp;
}
}
return result;
}
//生成随机数组
int randSize;
template <class T>
T* creatArr(int size, int value){
srand((unsigned)time(NULL));
randSize = rand()%(size+1)+1;
T *arr = new T [randSize];
for(int i = 0; i < randSize; i++){
arr[i] = rand()%(value+1) - rand()%(value+1);
cout << arr[i] << " ";
}
cout << endl;
return arr;
}
//拷贝数组
template<class T>
T* copyArr(T arr[], int size){
T* newArr = new T [size];
for(int i = 0; i < size; i++){
newArr[i] = arr[i];
}
return newArr;
}
int main(){
//-4,7,6,14,1,-9,-8
int testTime = 5000;
while(testTime--){
int *arr = creatArr<int>(10, 20);
int *arr2 = copyArr(arr, randSize);
if(insertSort(arr, randSize) == mergeSort(arr2, randSize)){
cout << "Nice!" << endl;
}else{
cout << "Error!" << endl;
cout << insertSort(arr, randSize) << " " << mergeSort(arr2, randSize) << endl;
break;
}
}
}