问题描述:
给定一个由整数组成的集合,集合中的整数各不相同,现在要将他们分为两个子集合,使得这两个子集合的并为原集合,交为空集,同时在两个子集元素个数n1 与 n2 之差的绝对值 | n1- n2 | 尽可能小的前提下,要求他们各自元素之和s1 和 s2 的差的绝对值 |s1 - s2 | 尽可能的大,求这个|s1 - s2 | 是多少,
第一个思路 sort 排序。
第二种思路就是随机选择 : 利用快速排序的性质,找到 第 n/2大的元素的位置就OK 时间复杂度O(N)。
可以写找第K个元素的位置,不需要排序(利用快速排序的原理):快速排序
#include<iostream>
#include<math.h>
#include<algorithm>
#include<ctime>
using namespace std;
const int maxn = 100010;
int A[maxn], n;
int randPartition(int A[], int left, int right){
int p = (round(1.0) * rand() /RAND_MAX * (right - left) + left);
swap(A[p], A[left]);
int temp = A[left];
while(left < right){
while(left < right && A[right] > temp) right--;
A[left] = A[right];
while(left < right && A[left] < temp) left++;
A[right] = A[left];
}
A[left] = temp;
return left;
}
int randSelect(int A[], int left, int right, int K){
if(left == right)
return A[left];
int p = randPartition(A, left, right);
int M = p - left + 1;
if(K == M){
return A[p];
}
if( K < M){
return randSelect(A, left, p - 1, K);
}else{
return randSelect(A, p + 1, right, K - M);
}
}
int main (){
srand((unsigned) time(NULL));
int sum = 0, sum1 = 0;
scanf("%d", &n);
for(int i = 0; i < n; i++){
scanf("%d", &A[i]);
sum += A[i];
}
randSelect(A, 0, n-1, n/2);
for(int i = 0; i < n/2; i ++){
sum1 += A[i];
}
printf("%d", (sum - sum1) - sum1);
return 0;
}