leetCode(30):Sort Colors

最新推荐文章于 2015-08-09 22:01:01 发布

朱传林

最新推荐文章于 2015-08-09 22:01:01 发布

阅读量639

点赞数

分类专栏： leetCode LeetCode

本文链接：https://blog.csdn.net/Walker19900515/article/details/46736995

版权

leetCode 同时被 2 个专栏收录

56 篇文章 0 订阅

订阅专栏

LeetCode

55 篇文章 1 订阅

订阅专栏

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

void sortColors(vector<int>& nums)
{//排序时间复杂度O(n)
	int length=nums.size();
	int zero=0;
	int two=length-1;
	while(zero<length && nums[zero]==0)
	{//找到第一个不为0的数的下标
		zero++;
	}
	while(two>=zero && nums[two]==2)
	{//找到倒数第一个不为2的数的下标,zero前的数都是0
		two--;
	}

    for(int i=zero;i<=two;++i)
    {
    	if(nums[i]==0)
    	{//如果为0，和zero指向的数交换，zero前移
    		nums[i]=1;//zero指向的值肯定是1，首先不可能为0，其次，i遍历过的地方，2都已经换到最后
    		nums[zero]=0;
    		zero++;
    	}
    	else if(nums[i]==1)
    	{//如果为1，则跳过
    		continue;
    	}
    	else
    	{//如果为2，则和two指向的数交换，并更新two的值，
    		//和zero不同，two前面的数可能还未排序
    		nums[i]=nums[two];
    		nums[two]=2;
    		two--;
    		while(two>=i && nums[two]==2)
    		{
    			two--;
    		}
    		i--;//先向后退一步，因为two指向的数还未进行排序
    	}
    }
}