无序整数数组中找第k大的数

方法一：基于快排

/*
基于区间快排的第K小算法 ,输出a[k-1]即可，O(n*logn)；每次只对后半部分递归便可把复杂度降到O(n) 
主要思路是每次随机在数组中选取一个元素p，利用这个元素将数组分成两部分，比p小的元素在分好的数组左边，p和比p大的元素在数组右边，
根据k值选择在数组左半或者右半部分继续递归执行查找。
*/
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

void swap(int *p, int *q)
{
    int t;
    t = *p;
    *p = *q;
    *q = t;
}

int findNumberK(vector<int> &vec, int k, int from, int to)
{
    int roll = vec[from], middle = 0, j = from;//roll是指标杆值 
    for(int i = from+1 ; i<= to ; i++)
    {
        if(vec[i] < roll)
        {
            j++;//roll左边每次被换的元素 的下标 
            swap(&vec[i], &vec[j]);
            middle++;
        }
    }
    swap(&vec[from], &vec[j]);
    if(middle == k -1 )
        return roll;
    else if (middle < k - 1)
    {
        return findNumberK(vec, k - middle - 1, j + 1, to);
    }
    else
        return findNumberK(vec, k, from, j - 1);
}

int main()
{
    vector<int> vec;
    int temp, k;

    cout << "input data(以0结束):" << endl;
    cin >> temp;

    while(temp != 0)
    {
        vec.push_back(temp);
        cin >> temp;
    }

    cout << "input K: " << endl;

    cin >> k;

    int re = findNumberK(vec , k, 0 ,vec.size() - 1);

    cout << "Test Result: "  << re << endl;

    sort(vec.begin(), vec.end(), less<int>());

    cout << "real Result: "  << vec[k-1] << endl;
    //while(1);
    return 0;
}