算法基础例四荷兰国旗问题

荷兰国旗问题

荷兰国旗问题： 给定一个数组arr，和一个数num，请把小于num的数放在数组的左边，等于num的数放在数组的中间，大于num的数放在数组的右边。
要求 额外空间复杂度O(1)，时间复杂度O(N)
解题思路：假设把数组分成三部分，小于num的区域，等于num的区域，大于num的区域。小于num的区域右边的前一个下标记为Less，等于num的区域最右边的下标记为cur，大于num的区域的左边的前一个下标记为more。
数组从0到N-1进行遍历，若当前元素小于num，则当前元素与小于num的区域最右边的数即Less所指的数进行交换，less++，cur++。若当前元素等于num，则当前元素不变，cur++。若当前元素大于num，则当前元素与大于num区域的左边一个元素（这里为待定区域）进行交换，并对交换过来的数与num进行比较。轮流下来依次比较，当cur与more等于时，停止。
代码如下：

# include <stdio.h>
void swap(int arr[], int i, int j)  //数组arr中第i个元素与第j个元素进行交换 
{
	int swap_temp = arr[i];
	arr[i] = arr[j];
	arr[j] = swap_temp;	
} 
void partition(int arr[], int left, int right, int num)
{
	int less = left - 1;
	int more = right +1;
	int cur = left;
	int i = 0;
	while( cur < more)   //结束条件是cur与more相等 
	{
		if(arr[cur] < num)    //当前元素小于num，与小于区域交换 
		{
			swap(arr, ++less, cur++);
		}
		else if(arr[cur] > more) //当前元素大于num，与大于区域交换并再次比较 
		{
			swap(arr, --more, cur++);
		}
		else
		{
			cur++;
		}
	}
	
	for(i = 0; i < (right - left + 1 ); i++)
	{
		printf("%d\t",arr[i]);
	}
} 
int main(void)
{
   int a[] = {500, 22, 53, -158, 4, 86, 32, 33, 26, -70, 135, -34, 32, 543, 2500};
  //  int a[] = {1,3,4,2,5,6};
    int n;  //存放数组a中元素的个数
    n = sizeof(a) / sizeof(a[0]);  /*a[0]是int型, 占4字节*/
    partition(a,0,n-1,100);
    printf("\r\n");
    return 0;
}