快排 Partition 写法

有两种写法:

第一种

int partition(vector<int>&arr, int low, int high)
{
    int pivot = arr[low];//选第一个元素作为枢纽元
    while(low < high)
    {
        while(low < high && arr[high] >= pivot)high--;
        arr[low] = arr[high];//从后面开始找到第一个小于pivot的元素，放到low位置
        while(low < high && arr[low] <= pivot)low++;
        arr[high] = arr[low];//从前面开始找到第一个大于pivot的元素，放到high位置
    }
    arr[low] = pivot;//最后枢纽元放到low的位置
    return low;

第二种

int partition(vector<int>&arr, int low, int high)
{
    int pivot = arr[high];//选最后一个元素作为枢纽元
    int location = low-1;//location指向比pivot小的元素段的尾部
    for(int i = low; i < high; i++)//比枢纽元小的元素依次放在前半部分
       if(arr[i] < pivot)
           swap(arr[i], arr[++location]);
    swap(arr[high], arr[location+1]);
    return location+1;