随机选择算法

随机选择算法

求浮点型的四舍五入，用round()函数，包含头文件<cmath>中。

RAND_MAX是C中stdlib.h中宏定义的一个字符常量：

#define RAND_MAX Ox7FFF

其值最小为32767,最大为2147483647

通常在产生随机小数时可以使用RAND_MAX。

rand()*1.0/RAND_MAX这个数就是产生了一个0-1的浮点数数据。

问题描述：

给定一个由整数组成的集合，集合中的整数各不相同，现将其分为两个子集合，使这两个子集合并为原集合，交为空集，在两个子集合元素个数n1，n2，差|n1 - n2|尽可能小时，要求它们各自元素和|s1 - s2|尽可能大。

#include<cstdio>
#include<cstdlib> 
#include<iostream>
#include<ctime>
#include<algorithm>
using namespace std;
const int maxn = 100010;
int A[maxn], n;

int randPartition(int A[], int left, int right){
	//生成[left, right]内的随机数p 
	int p = rand()/RAND_MAX*(right-left) + left;

	swap(A[p], A[left]);//交换A[p]和A[left]
	//执行原Partition函数的划分过程
	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);//划分后主元的位置为p 
	int M = p - left + 1; //A[p]是A[left, right]中的第M大 
	
	if(K == M) return;    //找到第K大的数 
    else if(K < M)//第K大的数在主元左侧 
	    randSelect(A, left, p - 1, K);//往主元左侧找第K大 
	else     //第k大的数在主元右侧 
	    randSelect(A, p + 1, right, K - M); //往主元右侧找第K-M大 
}
int main(){
	srand((unsigned)time(NULL));//初始化随机数种子
	//sum和sum1记录所有整数之和，与切分后前n/2个元素之和
	int sum = 0, sum1 = 0, i;
	cin >> n;//整数个数
	for(i = 0; i < n; i++){
		cin >> A[i];//输入整数 
		sum += A[i];//累计所有整数之和 
	}
	randSelect(A, 0, n-1, n/2);//寻找第n/2大的数，并进行切分 
	for(i = 0; i < n/2; i++){
		sum1 += A[i];//累计较小的子集合中元素之和 
	}
	cout << (sum - sum1) - sum1;//求两个子集合的元素和之差
	return 0;                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                  
}