算法导论 寻找第i小元素 9.2

PS1：如果单纯为做出这道题那么这个代价是O(nlgn)，通过排序就可以了。 这里讨论的是O(n)的算法。那么来分析一下这个算法是如何做到O(n)的，算了不分析了，这个推到看起来太麻烦了。其实我想知道的只是结论而已，想感叹的也是partition真是特别好用啊！！！！

代码：

#include<iostream>
#include<ctime>

using namespace std;

int size = 10;

void Swap(int &a, int &b)
{
    int c = a;
    a = b;
    b = c;
}

void RandomizedSwap(int a[], int p, int r)
{
    srand((int)time(0));
    int t = rand() % (r - p + 1)+p;
    Swap(a[p], a[r]);
}

int RandomizedPartition(int a[], int p, int r)
{
    RandomizedSwap(a, p, r);
    int x = a[r];
    int i = p - 1;
    for (int j = p; j < size - 1; j++)
    {
        if (a[j] < x)
        {
            i = i + 1;
            Swap(a[j], a[i]);
        }
    }
    Swap(a[i+1], a[r]);
    return i + 1;
}

int RandomizedSelect(int a[], int p, int r, int i)
{
    if (p == r)
        return a[p];
    else
    {
        int q = RandomizedPartition(a, p, r);
        if (q -p+1 == i)
            return a[q];
        else
        {
            if (q -p +1 < i)
                return RandomizedSelect(a, q + 1, r, i-q + p-1);
            else
                return RandomizedSelect(a, p, q - 1, i);
        }
    }
}

int main()
{
    int a[] = { 16, 4, 10, 14, 7, 9, 3, 2, 8, 1 };
    cout << RandomizedSelect(a, 0, 9, 9) << endl;
}