荷兰国旗问题 两种解法

问题介绍

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同 颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

解决方法

1.简单模拟实现。

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <time.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <stack>
#include <vector>
#include <string>
using namespace std;
#pragma warning(disable:4996)   
//这是我常用的头文件，实际可能用不了这么多。但都写上。

//该方法的意思是让这三个变量表示颜色的最后一个的下标。
int main() {
	int n = 0;
	cin >> n;
	int* a;
	a = new int[n];
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	int red = -1;
	int white = -1;
	int blue = -1;
	for (int i = 0; i < n; i++) {
		if (a[i] == 2) blue++;
		if (a[i] == 1) { blue++; white++; }
		if (a[i] == 0) { blue++; white++; red++; }
	}	
	for (int i = 0; i <=red; i++) {
		cout << 0<<" ";
	}
	for (int i = red+1; i <= white; i++) {
		cout << 1 << " ";
	}
	for (int i = white+1; i <=blue; i++) {
		cout << 2 << " ";
	}
	delete[]a;
}

2.利用快速排序的思想

快速排序利用的是分区域递归，这里设置前部，中部，后部三部分。分别对应0，1和2。主要思想是对“中部”进行操作。
设置三个“指针”。low指向第一位。high指向最后一位。还有一个遍历指针i，从头开始遍历。若遇到1，则i后移一位，若遇到0，则将i的值与low的值互换，i后移一位，low后移一位，若遇到2,则将i的值与high的值互换，则将i的值与high互换，high前移一位，i位置不变，因为新换过来的值可能是0，那就需要再换一次。
那为啥遇到0时不这么操作呢？因为0是要放到前面的，放在前面的一定是已经“排序好的”，因为是从前面开始遍历。

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <time.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <stack>
#include <vector>
#include <string>
using namespace std;
#pragma warning(disable:4996)

int main() 
{
	int n = 0;
	cin >> n;
	int* a;
	a = new int[n];
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	int low = 0;
	int high = n - 1;
	for (int i = 0; i < n&&i<=high;) {
		if (a[i] == 1) i++;
		else if(a[i]==0)
		{
			int temp = a[i];
			a[i] = a[low];
			a[low] = temp;
			i++;
			low++;
		}
		else if(a[i] == 2) {
			int temp = a[i];
			a[i] = a[high];
			a[high] = temp;
			high--;
		}
	}
	for (int i = 0; i < n; i++) {
		cout << a[i] << " ";
	}
	delete[]a;
	return 0;
}

好了，这是荷兰国旗问题的两种解法，谢谢阅读。