1113. Integer Set Partition (25)

题目链接

这道题本来是考类快速选择的算法的，因为题目数据太弱，导致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;
}