随机选择算法
求浮点型的四舍五入,用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;
}