一個長度爲n的數組，其每個元素的值爲1、2、3三個數之一，將數組排序



要求O(n)且不使用輔助空間，還要有一定的擴展性

#include <stdio.h>

void swap(int *a,int *b)
{
	int tmp = *a;
	*a = *b;
	*b = tmp;
}

void sort(int a[],int len)
{
	int i = 0;//從前向後指向第一個不爲1的數
	int j = 0;//從前向後指向第一個不爲2的數
	int k = len-1;//從後向前指向第一個不爲3的數

	while (i <= j && j <= k && i <= k)
	{
		//找到第一個不爲1的數
		while (a[i] == 1)
		{
			i++;
		}
		//找到第一個不爲2的數
		while (a[j] == 2)
		{
			j++;
		}
		//從後向前找到第一個不爲3的數
		while (a[k] == 3)
		{
			k--;
		}

		//第一個和最後一個比較
		if (i <= k && a[i] == 3 && a[k] == 1)
		{
			swap(&a[i],&a[k]);
			i++;
			k--;
		}
		else if (j <= k && a[j] == 3 && a[k] == 2)
		{
			swap(&a[j],&a[k]);
			j++;
			k--;
		}
		else if (i <= j && a[i] == 2 && a[j] == 1)
		{
			swap(&a[i],&a[j]);
			i++;
			j++;
		}
		else if (i < j && j < k)
		{
			a[i++] = 1;
			a[j++] = 2;
			a[k--] = 3;
		}

		if (i > j)
		{
			j = i;
		}
	}
}

int main()
{
	int a[] = {3,1,2,2,1,3,3,2};
	int i = 0;
	int len = sizeof(a)/sizeof(a[0]);
	sort(a,len);
	for (; i < len; i++)
	{
		printf("%d ",a[i]);
	}

	printf("\n");
	return 0;
}