数据结构-前正后负

 (2018吉林大学941)一个长度为 n 的数组由负数、0、正数组成。编写函数，将其重新排序为前段都是负数，后段均为非负数的结构。要求时间复杂度为 O(n)。  

 

#include<iostream>
#include<stdlib.h>
using namespace std;

void SortArray(int data[], int n)
{
	int head = 0, rear = n - 1, count = 0; 
	/*
		定义两个“指针”，在head之前（包括head）
		都是小于0数，rear之后（不包括rear）
		都是大于0的数
	*/ 
	int tmp;
	for(int i = 0; count < n ; count++)
	{
		/*
			判断当前元素
			--	如果是正数，则与rear所指元素交换，rear前移，计数器+1，指针不加1（因为交换过来的rear没有判断过） 
			--	如果是负数，则与head所指元素交换，head后移，计数器+1，指针+1 (因为移过来的head一定是小于0的)
			--	如果是0，则计数器+1，指针+1  
		*/ 
		if(data[i] < 0)
		{
			tmp = data[head];
			data[head++] = data[i];
			data[i++] = tmp;
		}
		else if(data[i] > 0)
		{
			tmp = data[rear];
			data[rear--] = data[i];
			data[i] = tmp;
		}
		else
		{
			i++;
		}
	}
}

void PrintArray(int data[], int n)
{
	for(int i = 0; i < n; i++)
	{
		cout<<data[i]<<" ";
	}
	cout<<endl;
}

int main()
{
	int data[] = {1, 0, -5, -9, 1, 0, 4, -6};
	PrintArray(data, 8);
	SortArray(data, 8);
	PrintArray(data, 8);
	return 0;
}