这道题本来是考类快速选择的算法的,因为题目数据太弱,导致O(nlogn)的排序算法也能AC。
下面的代码可以以平均O(n)的时间复杂度找到给定数组中第k大(0<=k<n
)的元素,同时以该元素将数组左右分开。
int Kth_elem(int a[],int low,int high,int k){
swap(a[low],a[(low+high)/2]);
int pivot = a[low];
int low_temp = low,high_temp = high;
while(low<high){
while(low<high&&a[high]>=pivot) high--;
a[low] = a[high];
while(low<high&&a[low]<=pivot) low++;
a[high] = a[low];
}
a[low] = pivot;
if(low==k) return a[low];
else if(low>k)
return Kth_elem(a,low_temp,low-1,k);
else return Kth_elem(a,low+1,high_temp,k);
}
其中中枢值pivot的选择非常重要,我选择的是数组中间的元素,还可以使用随机数法选择,尽量不要选择数组的首尾元素。
另外,C++头文件algorithm
中预置了一个相同功能的函数,nth_element - C++ Reference
nth_element(a,a+k,a+n);
时间复杂度也是O(n)。
下面是直接使用排序的代码。
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int maxn = 100001;
int a[maxn];
int main(){
int n,s1 = 0,s2 = 0;
cin >> n;
for(int i = 0;i < n;i++){
scanf("%d",&a[i]);
}
sort(a,a+n);
// nth_element(a,a+n/2,a+n);
for(int i = 0;i < n;i++){
if(i<n/2) s1 += a[i];
else s2 += a[i];
}
cout << n%2 << ' ' << s2-s1 << endl;
return 0;
}