选择问题001

#include<iostream>
#include<cmath>
using namespace std;
#define NUM 1001
int a[NUM];
int select(int left, int right, int k)//left:数组起始下标（本题从0开始）；right：终点下标；k:数组中第k小的数。
{
    int i = left;//i相当于指针（从左向右）
    int j = right;//j相当于指针（从右向左）
    int pivot = a[left];//选第一个数作为分界数据，将比它小的数放在它左边，大的放在右边。
    while (true)
    {
        while (a[i] < pivot)//如果a[i] < pivot,i指针就不停的向右移动
        {
            i++;
        }while (a[j] > pivot)//如果a[j] > pivot, j指针就不停的向左移动
        {
            j--;
        }    
        if (i == j) break;//如果i==j说明将pivot放入了最中间的位置。
        else
        {
            swap(a[i], a[j]);//走不动了，交换一下a[i]和a[j]。
        }
        if (a[i] == a[j])j--;//如果遇到两数相等，默认j所指的大一点，j--。如(2 1 2 5 3) 如果没这一步会死循环。
    }
    //nleft表示pivot左边的元素个数。nleft=j-left。
    if (j - left == k - 1)//pivot的位置等于左边的元素个数+1.这时正好直接返回pivot。
    {
        return pivot;
    }
    if (j - left < k - 1)//右边多了，从j+1开始向右寻找第k-（nleft+1)小的数。（因为已经找到了前nleft+1小的数了。）
    {
        return select(j + 1, right, k - (j - left + 1));
    }
    else//说明第k小的数在左边集合中，从第一个元素开始找，找到第k小的元素就好。
    {
        return select(left, j - 1, k);
    }

}
int main()
{
    int n,x;
    cout << "请输入有几个数" << endl;
    cin >> n;
    cout << "请输入要寻找第几小的数" << endl;
    int k;
    cin >> k;
    for (int i = 0; i < n; i++)
    {
        cout << "请输入第" << i+1 << "个数的值是多少" << endl;
        cin >> x;
        a[i] = x;
    }
    cout<<"第"<<k<<"小的数是" << select(0, n - 1, k);
}