数组的partition调整/c++

做在线编程题时遇到了这个部分调整的题目，思路来自CSDN博主「BUPT-WT」的原创文章，用C++编写了一下。
原文链接：https://blog.csdn.net/weixin_41362649/article/details/95043654

题目

原题目：
　　给定一个有序数组arr，调整arr使得这个数组的左边部分没有重复元素且升序，而不用保证右半部分是否有序。
　　例如，arr = [1, 2, 2, 2, 3, 3, 4, 5, 6, 6, 7, 7, 8, 8, 8, 9]，调整之后arr = [1, 2, 3, 4, 5, 6, 7, 8, 9……]

补充题目：
　　给定一个数组arr，其中只可能含有0,1,2三个值，请实现arr的排序。
　　另一种问法：有一个数组，其中只有红球、蓝球和黄球，请实现红球全放在数组的左边，蓝球放在中间，黄球放在右边。
　　另一种问法：有一个数组，再给定一个值k，请实现比k小的数都放在数组的左边，等于k的数都放在数组的中间，比k大的数都放在数组的右边。

基本思路

原问题：
　　原数组有序，只要将重复的元素往后放置。变量left表示数组中无重复且升序的区域范围，初始是为0。从位置1开始遍历数组，如果arr[i] != arr[left]，说明该元素可以加到左区域，这时候将左区域扩大一个单位用来放置arr[i]，令left+1，然后将arr[left]和arr[i]交换；如果arr[i] == arr[left]，说明该元素之前已经出现过，不需要加入到左区域。

补充问题：
变量left和right分别表示左右两个区域，arr[0…left]表示都为0的区域，arr[right…N-1]表示都为2的区域。初始时令left = -1，right = len(arr)，表示区域中无任何元素。令index = 0，使用这个变量做从左到右的遍历，直到右区域的起点。

如果arr[index] == 0，将左区域扩大一个单位来放置该元素，把arr[left+1]和arr[index]交换即可。同时令left、index加1，继续遍历下一个位置。

如果arr[index] == 2，将右区域扩大一个单位来放置该元素，把arr[index]和arr[right-1]交换即可，令right减1，同时，index不变，继续从该位置遍历。

如果arr[index] == 1，直接令index加1，继续遍历下一个位置。

原问题代码

#include<iostream>
using namespace std;

void sortLeft(int * const arr,int N) 
{
	if (N > 2)
	{
		int left = 0;
		for (int i = 1; i < N; i++)
		{
			if (arr[left]<arr[i]) 
				swap(arr[++left], arr[i]);
		}
	}
}

int main()
{
	int N;
	cin >> N;
	int *arr = new int[N];
	for (int i = 0; i<N; i++)
	{
		cin >> arr[i];
	}
	
	sortLeft(arr, N);
	for (int i = 0; i<N; i++)
	{
		cout << arr[i ]<<" ";
	}
	delete [] arr;
	system("pause");
	return 0;
	
}

补充问题代码

#include<iostream>
using namespace std;

void sortLeft(int * const arr, int N, int k) //k为中间值
{
	if (N > 2)
	{
		int left = -1;
		int right = N;
		for (int i = 0; i < right; i++)
		{
			if (arr[i]<k)
				swap(arr[++left], arr[i]);
			else if (arr[i]>k)
			{
				swap(arr[--right], arr[i]);
				i--;
			}				
		}
	}
}

int main()
{
	int N;
	cin >> N;
	int *arr = new int[N];
	for (int i = 0; i<N; i++)
	{
		cin >> arr[i];
	}

	sortLeft(arr, N, 1);

	for (int i = 0; i<N; i++)
	{
		cout << arr[i] << " ";
	}
	delete [] arr;
	return 0;

}