/*
 ******************************************************************************
 * file name  : templ_sort.h
 * author     : Hung Q. Ngo
 * description: implementation a template version of randomized quick sort
 *              and insertion sort
 ******************************************************************************
 */

#include <vector>
#include <cstdlib>  // for rand() and srand()
#include <ctime>    // for time()

/**
 * -----------------------------------------------------------------------------
 * Template Insertion Sort
 * for each i from 1 to n-1, insert vec[i] to the right place in the prefix
 * -----------------------------------------------------------------------------
 */
template <typename Item_Type>
void insertion_sort(std::vector<Item_Type> &vec) {
    Item_Type temp, j;
    for (int i=1; i<vec.size(); i++) {
        temp = vec[i];
        j = i-1;
        // assumes compiler knows how to do '>' with Item_Type
        while (j >= 0 && vec[j] > temp) {
            vec[j+1] = vec[j];
            j--;
        }
        vec[j+1] = temp;
    }
}

/**
 * -----------------------------------------------------------------------------
 * The Randomized Quick Sort algorithm
 * - pick a random pivot, that's all
 * -----------------------------------------------------------------------------
 */
template <typename Item_Type>
void rqs_with_range(std::vector<Item_Type> &vec, int p, int q) {
    if (p >= q) return;

    // pick a random pivot
    int m = std::rand() % (q-p+1);
    std::swap(vec[q], vec[p+m]); 

    // partition
    int r=p-1;
    for (int j=p; j<q; j++)
        if (rand() % 2 == 0) {
            if (vec[j] < vec[q])
                std::swap(vec[++r], vec[j]);
        } else { 
            if (vec[j] <= vec[q])
                std::swap(vec[++r], vec[j]);
        }

    std::swap(vec[++r], vec[q]);

    // sort the left & the right parts
    rqs_with_range(vec, p, r-1);
    rqs_with_range(vec, r+1, q);
}

/**
 * -----------------------------------------------------------------------------
 * Wrapper to the randomized quick sort algorithm
 * -----------------------------------------------------------------------------
 */
template <typename Item_Type>
void randomized_quick_sort(std::vector<Item_Type> &vec) {
    std::srand(static_cast<unsigned int>(std::time(0)));
    rqs_with_range(vec, 0, vec.size()-1);    
}