《算法笔记》-两个子集合个数差最小，子集和的差最大

思路：

1、用快速排序思想，找到中位数并排好，令中位数左边均小于该数，右边均大于该数。

2、此时得到两个集合，s1，s2。易知，s1 中每个数均小于中位数，s2中每个数均大于中位数。即s1 < 中位数 < s2。

ps：当数组个数为偶数时，s1的个数与s2集合中的数字个数相等，所谓的中位数其实是s2中的第一个数。如s1 ={1,2,3}, s2 = {4,5,6}

当数组个数为奇数时，s1的个数比s2的个数少一个，中位数归属s2。如s1= {1,2,3}，s2={4,5,6,7}，中位数是4。

#include<iostream>
#include<vector>
using namespace std;
vector<int> a;
int partition(int left, int right){//快速排序算法的一部分，使当前位插入到合适的位置，使其左边均小于temp，右边均大于temp 
	int temp = a[left];
	while(left < right){
		while(left < right && a[right] > temp) right--;//右指针向左移动，直至找到一个 a[right] <= temp的位置 
		a[left] = a[right];
		while(left < right && a[left] <= temp) left++;//左指针向右移动，直至找到第一个a[left] > temp的位置 
		a[right] = a[left];
	}
	a[left] = temp;
	return left;
}
void get(int k, int left, int right){//根据思路可得，k选取中位数。此时，中位数左边为小集合，右边为大集合，此时差一定是最大的。 
	int m = partition(left, right);
	if(k == m + 1){//由于数组下标从0开始，则下标i，实际上是第i+1位数。 
		int ans = 0;
		for(int i = 0; i < a.size()/2; ++i) //令大集合的数减去集合的数。注意a.size()为奇数时， a.size()/2该为未被访问到，且属于大集合 
			ans += a[a.size()- 1- i] - a[i];
		if(a.size() & 1) ans += a[a.size()/2];//当数组个数为奇数时，a.size()/2该位数属于大集合，且未被访问到 
		cout << ans;
		return;
	}
	if(k > m + 1) return get(k, m + 1, right);
	return get(k, left, m - 1);
}
int main(){
	int n;
	cin >> n;
	a.resize(n);
	for(int i = 0; i < n; ++i)
		cin >> a[i];
	get(n/2 + 1, 0, n - 1);
}

输入：

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

输出：

80

输入：

6
2 3 5 7 9 12

输出：

18