荷兰国旗算法

给出一个无序的数列，给出一个划分值，要求将数列中小于此数的全部放在左边的小于区，等于的放在中间的等于区，大于的放在右边大于区。

数列为vector<int> arr

此时定义三个指针，一个指向小于区，一个指向大于区，一个指向当前遍历数。一开始小于区处于-1的位置，大于区处于arr.size()处，当前数处于下标0处。于是开始下面的流程：

1）若当前数小于划分数，则将当前数与小于区的下一个数交换，并将小于区扩大一格，当前数向前跳一个。

2）若当前数等于划分数，当前数直接向前跳一个。

3）若当前数大于划分数，则将当前数和大于区的前一个进行交换，大于区向前扩大一格，当前数保持不动

直到当前数的下标与大于区的下标相等后退出。

 

#include<iostream>
#include<vector>
using namespace std;


int*partition(vector<int> &arr, int L, int R);
void Swap(vector<int> &arr, int i, int j);
int main()
{
	vector<int> arr = { 9,5,4,8,90,4,3,5,78,9,56,3,4,5,6,5 };
	int *a = partition(arr, 0, arr.size() - 1);
	for (int i = 0; i < arr.size(); i++)
	{
		cout << arr[i] << endl;
	}
	for (int i = 0; i < 2; i++)
	{
		cout << a[i] << " ";
	}
	cout << endl;
}

int *partition(vector<int> &arr, int L, int R)
{
	int less = L - 1;
	int more = R;
	int index = L;
	int part = arr[R];
	while (index < more)
	{
		if (arr[index] < part)
		{
			Swap(arr, ++less, index++);
		}
		else if (arr[index] > part)
		{
			Swap(arr, --more, index);
		}
		else
		{
			index++;
		}
	}
	Swap(arr, more, R);
	int *result = new int[2];
	result[0] = less + 1;
	result[1] = more;
	return result;
}

void Swap(vector<int> &arr, int i, int j)
{
	int temp = arr[i];
	arr[i] = arr[j];
	arr[j] = temp;
}