[置顶] 从一位数组中提取最小k个元素

#include <iostream>
#include <cstdlib>
#include <vector>

using namespace std;
/*
 * 从n个整型数组中提取前k个最小整数
 * 方法一：利用快速排序进行递归划分
*/
const int Max = 100;   // 数组最大容量
class Solution {
private:
	vector<int>array;  // 输入的数组 
	vector<int>kmin;   // 提取的k个最小值的数组
    /*   
	 * 对数组进行划分，返回“中值”位置
	*/
	int Adjust(vector<int>&arr, int low, int high) {
		int i, j;
		i = low;
		j = high;
		int tmp = arr[low];
		while(i < j) {
			while(i < j && arr[j] > tmp) j--;
			if(i < j) arr[i++] = arr[j];

			while(i < j && arr[i] < tmp) i++;
			if(i < j) arr[j--] = arr[i];

		}
		arr[i] = tmp;
		return i;
	}
    /*
	 * 利用上面的调整函数进行查找
	*/
	void JustDoIt(vector<int>&arr, int low, int high, int k) {
		int i;
		int mid = Adjust(arr, low, high);
		static bool find;     // 用于递归截断
		/*
		 * 找到解
		*/
		if(!find && (mid + 1 == k || mid == k)) {
			find = true;
			for(i = 0;i < k;i++) {
				kmin.push_back(arr[i]);
			}
			return;
		}
		/*
		 * 递归处理 
		*/
		else if(!find && k > mid + 1) {
			JustDoIt(arr, mid + 1, high, k);
		}
		else if(!find) {
			JustDoIt(arr, low, mid - 1, k);
		}
	}

public:
	Solution(int n) {
		if(n < 0 || n>= Max) {
			cerr<<"参数不合法!"<<endl;
			exit(1);
		}
		int i;
		int e;
		for(i = 0;i < n;i++) {
			cout<<"请输入第"<<i<<"个整数:";
			cin>>e;
			if(!cin.good()) exit(1);
			array.push_back(e);
		}
	}
	void JustDoIt(int k) {
		if(k <= 0 || k > array.size()) {
			cerr<<"参数不合法!"<<endl;
			exit(1);
		}
		JustDoIt(array, 0, array.size() - 1, k);
	}
	void show()  {
		vector<int>::iterator it;
		for(it = kmin.begin(); it != kmin.end(); it++) {
			cout<<*it<<endl;
		}
	}
};

void main() {
	Solution s(10);
	s.JustDoIt(3);
	s.show();
}