划分（Partition）算法实践

一、二划分算法（2-way-partition algorithm）
在数据结构中，快速排序quicksort的核心是对无序向量进行快速划分，选取一个元素作为轴点(pivot)对向量进行划分，确保比轴点大的元素在轴点之后，比轴点小的元素在轴点之前，将原向量划分为两个子向量。
其算法基本思想为：
算法思想：

1. 取一元素为轴点(pivot)，不妨取首元素为轴点，并将轴点的值备份；
2. 从向量的起始(low)和末尾(high)同时进行扫描；
3. 若nums[high] < pivot, 将其换到nums[low];若nums[low] > pivot,将其换到nums[high];
4. 当low与high位置重合后，将备份的pivot值填回nums[low].

实现代码如下：

int partition(vector<int>& nums, int low, int high) {
    // 任取一元素与首元素交换，以交换后的首元素作为基准
    int r = low + rand() % (high - low + 1);
    int tmp = nums[low];
    nums[low] = nums[r];
    nums[r] = tmp;
    int pivot = nums[low];
    // 由向量两端向中部扫描
    while (low < high) {
        while (low < high && pivot <= nums[high])
            high--;
        nums[low] = nums[high];
        while (low < high && pivot >= nums[low])
            low++;
        nums[high] = nums[low];
    }
    nums[low] = pivot;
    return low;
}

显然，该划分算法的时间复杂度是O(n)，空间复杂度为O(1).

二、三划分算法（3-way-partition algorithm）
先来看这样一个问题：
给定一个包含数字 1，2，3 的数组：[2，3，1，2，3，3，2，2，1，1，3， 2，1，3，1]，请把数字 1 放在数组左边，数字 3 放到数组右边，数字 2 位于 数组中间。 要求设计一个时间性能为 O(n)的算法。
其实不难看出，这是一个典型的三划分问题，其解决思路与Leetcode中的颜色分类问题基本相似：
给一个数组包含n个物体，有蓝色、红色和白色三种颜色，把他们分类并按照红、白、蓝的顺序排列，我们用0、1、2来表示红白蓝的颜色。（原题链接：Sort Colors）
下面分别给出具体的实现算法。
1.第一个问题具体实现代码：

#include <bits/stdc++.h>
#include<vector>
using namespace std;

void swap(int *a,int i,int j)   
{
	int tmp;
	tmp=a[i];
	a[i]=a[j];
	a[j]=tmp;
}

//三分类算法 
void Sort(vector<int>& t) {
    if (t.empty()) return;
    int i = 0, low = 0, high = t.size() - 1;
    while (i <= high) {
        if (t[i] == 2) i++;
        else if (t[i] == 1) swap(t[i++], t[low++]);
        else swap(t[i], t[high--]);
    }
}

int main()  
{ 
    vector<int> vec {2,3,1,2,3,3,2,2,1,1,3,2,1,3,1};
	Sort(vec);
	for(int i=0;i<vec.size();i++)
		cout<<vec[i]<<' ';
	cout<<endl;
	return 0;
}

2.Sort Colors问题算法实践：

// 3-way partition
void sortColors(vector<int>& nums) {
    if (nums.empty()) return;
    int i = 0, low = 0, high = nums.size() - 1;
    while (i <= high) {
        if (nums[i] == 1) i++;
        else if (nums[i] == 0) swap(nums[i++], nums[low++]);
        else swap(nums[i], nums[high--]);// not i++
    }
}

如果觉得本文对您有帮助，可以点个赞哦！以上内容若有侵权，请及时联系我删除。若您也对算法感兴趣，可以通过QQ：3451707815或3451707815@qq.com与我交流！谢谢您的支持！