《算法笔记》——4.7.3随机选择算法

题目：

给定一个整数（各不相同）组成的集合，现将其分成两个集合，要求两级集合交集为空，并集为整个集合同时要求两个自己和元素个数n1和n2的差的绝对值|n1-n2|尽可能小的前提下，要求它们各自元素之和S1和S2之差的绝对值|S1-S2|尽可能大。求|S1-S2|

思路：

1、最直接的想法就是从小到大排序，取前n/2个元素为一个集合，剩下的元素为另外一个集合，时间复杂度为O(nlogn).

2、更优算法（随机选择算法）：类似于随机快速选择排序算法。当对A[left，right]执行一次randparitaton后，主元左侧元素个数是确定的，且他们都小于主元。假设此时的主元是A[p],那么A[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大，往右侧递归就可以。

代码：

#include<stdio.h>
#include<math.h>
#include<time.h>
#include<algorithm>
using namespace std;
#define MAX 100010 
int A[MAX],n;

int randparitation(int A[],int left,int right){//选取随机主元，对区间[left,right]进行划分 
	int p=round(1.0*rand()/RAND_MAX*(right-left)+left);//选取[left,right]范围内的随机数 
	swap(A[left],A[p]);
	//以下部分为原来的paritation的代码 
	int temp=A[left];
	while(left<right){
		while(left<right&&A[right]>=temp){
			right--;
		}
		A[left]=A[right];//不满足while时交换 
		while(left<right&&A[left]<=temp){
			left++;
		}
		A[right]=A[left];//不满足while时交换 
	} 
	A[left]=temp;//将temp放到left和right相遇的地方
	return left;//由于left和right相等，故返回哪个都可以 
} 
void randselect(int A[],int left,int right,int k){//随机选择函数，选择[left,right]之间的第K大的数字，并进行切分 
	if(left==right) return;
	int p=randparitation(A,left,right);//划分后主元位置为p 
	int m=p-left+1; //A[p]是A[left,right]中的第M大
	if(m==k) return;//找到了第K大的数字 
	if(k<m){
		randselect(A,left,p-1,k);//往左侧找K大 
	} 
	if(k>m){
		randselect(A,p+1,right,k-m);//往右侧找地k-m大 
	}
} 

int main()
{
	srand((unsigned)time(NULL));//初始化随机种子
	int sum=0,sum1=0;//sum,sum1记录划分前的数字总和以及划分后的前n/2总和
	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;
}