一个长度为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;
}