链接:https://www.nowcoder.com/questionTerminal/edfe05a1d45c4ea89101d936cac32469
来源:牛客网
数组小和的定义如下:
例如,数组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 mergeSort(int a[],int n);
long sortProcess(int a[],int L,int R);
long merge(int a[],int L,int mid,int R);
int main()
{
int n;
cin>>n;
int a[n];
for(int i=0;i<n;i++)
cin>>a[i];
cout<<mergeSort(a,n)<<endl;
return 0;
}
long mergeSort(int a[],int n)
{
if(a==NULL||n<2)
return 0;
return sortProcess(a,0,n-1);
}
long sortProcess(int *a,int L,int R)
{
if(L>=R)
return 0;
int mid=L+((R-L)>>1);//相当于(L+R)/2,可有效避免溢出
return sortProcess(a,L,mid)+sortProcess(a,mid+1,R)+merge(a,L,mid,R);
}
long merge(int *a,int L,int mid,int R)
{
int help[R-L+1];//保存每次merge后的数组
memset(help, 0, sizeof(help));
int i=0;
int p1=L;
int p2=mid+1;
long rul=0;
while(p1<=mid&&p2<=R)
{
//设任意p2<m<R,当a[p1]<=a[p2]时,a[p1]<a[m]也显然成立,m的数量为 R-p2+1
rul+=a[p1]<=a[p2]?(R-p2+1)*a[p1]:0;
help[i++]=a[p1]<=a[p2]?a[p1++]:a[p2++];//每次选择小的数字存入数组
}
//左右两边的数量不是每次都相等,将剩下的元素也添加到help数组中
while(p1<=mid)
{
help[i++]=a[p1++];
}
while(p2<=R)
{
help[i++]=a[p2++];
}
for(i=0;i<R-L+1;i++)
a[L+i]=help[i];
return rul;
}