荷兰国旗问题

问题描述

给定一个数组a和一个数n，把小于等于n的数字放在数组的左边，大于n的数放在数组的右边，要求额外空间复杂度O(1)，时间复杂度O(n)

#include <iostream>
using namespace std; 

const int N = 100010;
int Arr[N];

void Sort(int *p, int nStart, int nEnd, int nFlag);

int main(int argc, char** argv) 
{
	int nLen;
	int nFlag;
	
	printf("Input All Numbers\n");
	scanf("%d",&nLen);
	
	printf("Input Numbers\n");
	for(int i = 0; i < nLen; i++)
	{
		scanf("%d",&Arr[i]);
	}
	
	printf("Input Number To Div\n");
	scanf("%d",&nFlag);
	
	Sort(Arr, 0, nLen-1, nFlag);
	
	for(int i = 0; i < nLen; i++)
	{
		printf("%d ",Arr[i]);
	}
	printf("\n");
	
	return 0;
}

void Sort(int *p, int nStart, int nEnd, int nFlag)
{
 	int nCur = nStart;//记录当前走到了数组的哪个位置
 	//nCur扫过的区域都是小于等于nFlag的 
	int nLess = nStart-1;//记录当前小于Flag的范围
	int nMore = nEnd+1;//记录当前大于Flag的范围
	
	while(nCur < nMore)
	{
		//如果当前的这个数 比 Flag小 
		if(Arr[nCur] < nFlag)
		{
			//当前的这个数 和 小于区域的下一个数交换 
			swap(Arr[nCur], Arr[nLess+1]);
			//当前指针 和 小于区域 都向后移动一位 
			nCur++; nLess++;	
		}
		//如果当前的数 比 Flag大 
		else if(Arr[nCur] > nFlag)
		{
			swap(Arr[nCur], Arr[nMore-1]);
			//大于区域向前移动一位 
			//nCur为什么不变呢 
			//因为不知道交换的这个数和Falg的大小关系 
			nMore--; 
		}
		//当前数字和Flag相等 那么不做任何处理 
		else
		{
			nCur++;	
		} 
	} 
}