75. Sort Colors(unsolved)

最新推荐文章于 2020-06-11 21:49:01 发布

gdmmzmj

最新推荐文章于 2020-06-11 21:49:01 发布

阅读量245

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode算法心得

本文链接：https://blog.youkuaiyun.com/gdmmzmj/article/details/66972824

leetcode算法心得专栏收录该内容

226 篇文章

订阅专栏

本文介绍了一种高效的算法，用于将一个包含红色、白色和蓝色对象的数组进行排序，使得相同颜色的对象相邻，并按照红、白、蓝的顺序排列。通过使用0、1、2分别代表红色、白色和蓝色，该算法利用三个指针red、i和blue来实现快速排序。

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.

解答：
设定一个red指针指向开头，一个blue指针指向结尾。然后用i遍历，如果遇到0
，那么与red交换，如果遇到2，那么与blue交换。如果遇到1，那么不用交换。应该注意到：
1.与blue交换后，i–
2. 与i是遍历到blue的。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int i=0;int red=0;int blue=nums.size()-1;
        if(nums.size()==1) return;
        for(int i=0;i<=blue;i++)
        {
            if(nums[i]==0)
                swap(nums[red++],nums[i]);
            else if(nums[i]==2)
                swap(nums[blue--],nums[i--]);
        }

    }
};