排颜色问题——数组 leetcode lintcode
问题描述:
给一个数组,并且数组里面元素的值只可能是0,1,2,然后现在把这个数组排序。
另外一种表述: 现有n个红白蓝三种不同颜色的小球,乱序排列在一起,请通过两两交换任意两个球,使得从左至右,依次是一些红球、一些白球、一些蓝球。
解题思路:
采用快速排序partition的思想,用两个指针将三种颜色间隔开。
下面引用此处内容 July 编程艺术系列中荷兰国旗问题。
一个前指针begin,一个中指针current,一个后指针end,current指针遍历整个数组序列,当
- current指针所指元素为0时,与begin指针所指的元素交换,而后current++,begin++ ;
- current指针所指元素为1时,不做任何交换(即球不动),而后current++ ;
- current指针所指元素为2时,与end指针所指的元素交换,而后,current指针不动,end-- 。
为什么上述第3点中,current指针所指元素为2时,与end指针所指元素交换之后,current指针不能动呢?因为第三步中current指针所指元素与end指针所指元素交换之前,如果end指针之前指的元素是0,那么与current指针所指元素交换之后,current指针此刻所指的元素是0,此时,current指针能动么?不能动,因为如上述第1点所述,如果current指针所指的元素是0,还得与begin指针所指的元素交换。
参考代码:
class Solution{ public: /** * @param nums: A list of integer which is 0, 1 or 2 * @return: nothing */ void sortColors(vector<int> &nums) { // write your code here int begin=0, cur = 0; int end = nums.size()-1; //0—红 1—白 2—蓝 while(cur <= end) { if(nums[cur] ==0) { swap(nums[begin], nums[cur]); begin++; cur++; }else if(nums[cur] == 1) { cur++; }else if(nums[cur] == 2) { swap(nums[cur], nums[end]); end--; } } } //交换函数 void swap(int &a, int &b) { int temp = a; a = b; b = temp; } };
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。