码迷,mamicode.com
首页 > 编程语言 > 详细

排颜色问题——数组 leetcode lintcode

时间:2015-06-08 09:45:17      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:快排partition   排颜色   数组   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;
    }
};



排颜色问题——数组 leetcode lintcode

标签:快排partition   排颜色   数组   leetcode   lintcode   

原文地址:http://blog.csdn.net/tommyzht/article/details/46408081

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!