高效技巧与算法——随机选择选择算法原理以及应用

原理类似于随机快速排序。

例题：从一个无序数组中求出第K大的数（数组中数各不相同）

思路1:先排序，然后直接去除第K个元素，但是需要O（nlogn）的时间复杂度

思路二：随机选择算法

先执行一次randPartition函数后，主元左侧的元素个数就是确定的，且它们都小于主元。

假设主元是A[p]，那么p就是A[left,right]中第p -left +1大的数。

不妨设 M表示第p-left+1，如果K==M，说明第K大的数就是主元A[p]；

如果K<M成立，就说明第K大的数载主元的左侧，即A[left…(p-1)]中的第K大，往左侧递归即可；

如果K>M成立，则说明第K大的数载主元右侧，即A((p+1)…right)中的第K-M大，往右侧递归即可。

算法以left==right作为递归边界。

参考代码：

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 randSelsect(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  randSelsect(A, left, p - 1, K);
    } else{
        return  randSelsect(A, p + 1, right, K - M);
    }
}

 

问题应用：

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

示例：

输入：

13 

1 6 33 18 4 0 10 5 12 7 2 9 3

输出：

80

//分集合，求集合之差最大
#include<cstdio>
#include <cstdlib>
#include<ctime>
#include <cmath>
#include<algorithm>

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 randSelsect(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  randSelsect(A, left, p - 1, K);
    } else{
        return  randSelsect(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];
    }
    randSelsect(A, 0, n - 1, n/2);

    for (int j = 0; j < n/2; ++j) {
            sum1 += A[j];
    }
    printf("%d\n", (sum -sum1) -sum1);
    return  0;
}