随机选择算法相关问题

随机选择算法相关问题

给定一个整数组成的集合，集合中的整数各不相同，现在要将它分为两个子集合，使得这两个子集合的并为原集合、交为空集，同时在两个子集合的元素个数n1与n2之差的绝对值|n1-n2|尽可能小的前提下，要求他们各自的元素之和S1和S2之差的绝对值|S1-S2|尽可能大。求|S1-S2|。
使用随机选择算法的原理就是，求这个整数集合的第n/2大，同时根据这个数把集合分为两个部分，使得其中一个子集元素都不小于这个数，另一个都大于这个数。基于随机快速排序算法思想。

#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <algorithm>
#include <cmath>

using namespace std;

const int maxn = 100010;
int A[maxn], n;

int randPartition(int A[], int left, int right) {
    int p = (int)round(1.0 * rand() / RAND_MAX * (right - left) + left);
    swap(A[left], A[p]);
    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;
}

void randSelect(int A[], int left, int right, int K) {
    if (left == right) return;
    int p = randPartition(A, left, right);
    int M = p - left + 1;
    if (M == K) return;
    if (K < M) {
        randSelect(A, left, p - 1, K);
    }
    else {
        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\n", (sum - sum1) - sum1);
    return 0;
}