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;
}