死亡拆分II

Problem Description

给定一个由整数组成的集合，集合中的整数各不相同，现在要将它分为两个子集合，使得这两个子集合的并为原集合、交为空集，同时在两个子集合的元素个数n1与n2之差的绝对值|n1-n2|尽可能小的前提下，要求它们各自的元素之和S1与S2之差的绝对值|S1-S2|尽可能大。

Input
每个输入文件中一组数据。

第一行一个正整数N（1<=N<=10000000），表示集合中的整数个数。第二行是用空格隔开的N个绝对值在6666666以内的整数。

Output
用空格隔开的两个整数，即所求的|n1-n2|与|S1-S2|。

Sample Input
5
4 2 1 3 5

Sample Output
1 9

Author
Shoutmon

Source
16浙大考研机试模拟赛

方法一：nth_element()

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
    vector<int> v;
    int n;
    scanf("%d",&n);
    v.resize(n);
    for(int i=0;i<n;++i)
        scanf("%d",&v[i]);
    int k=n/2;
    nth_element(v.begin(),v.begin()+k,v.end());
    long long sum=0;
    for(int i=0;i<k;++i)
        sum+=v[n-1-i]-v[i];
    n%2?printf("1 %lld\n",sum+abs(v[k])):printf("0 %lld\n",sum);
    return 0;
}

方法二：partition()

#include <cstdio>
#include <algorithm>
using namespace std;
int A[10000000+1],n;
void partition(int A[],int L,int R,int k)
{
    int mid=(L+R)/2;
    if(A[L]>A[mid]) swap(A[L],A[mid]);
    if(A[mid]>A[R]) swap(A[R],A[mid]);
    swap(A[mid],A[L]);
    int i=L,j=R;
    int pivot=A[i];
    while(i<j)
    {
        while(i<j&&A[j]>pivot) --j;
        A[i]=A[j];
        while(i<j&&A[i]<pivot) ++i;
        A[j]=A[i];
    }
    A[i]=pivot;
    if(i==k) return;
    else if(i<k) partition(A,i+1,R,k);
    else partition(A,L,i-1,k);
}
int main()
{
    // freopen("m.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;++i)
        scanf("%d",&A[i]);
    int k=n/2;
    partition(A,0,n-1,k);
    long long ret=0;
    for(int i=0;i<k;++i)
        ret+=A[n-i-1]-A[i];
    if(n%2==0) printf("0 %lld\n",ret);
    else printf("1 %lld\n",ret+abs(A[k]));
    return 0;
}