【leetcode】75. Sort Colors

最新推荐文章于 2020-10-07 20:01:40 发布

小do在努力

最新推荐文章于 2020-10-07 20:01:40 发布

阅读量394

点赞数

分类专栏：【leetcode】文章标签： leetcode one pass sort colors

本文链接：https://blog.youkuaiyun.com/zone_programming/article/details/49122835

版权

【leetcode】专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种颜色排序算法的两种实现方式：两遍扫描法和单遍扫描法。两遍扫描法首先统计每种颜色的数量，然后按顺序填充数组；单遍扫描法则使用两个指针分别标识红色和蓝色的位置，通过交换元素实现排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

第一种解法，统计red,white,blue的个数，然后再扫一遍进行置数，two pass

/**
 *  @author         johnsondu
 *  @time           11:26 14th Oct 2015
 *  @status         Accepted
 *  @strategy       two pass
 *  @problem        Set Matrix Zeroes
 *  @url            https://leetcode.com/problems/set-matrix-zeroes/
 */

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int color[3] = {0};
        for(int i = 0; i < nums.size(); i ++)
            color[nums[i]] ++;
        int cnt = 0;
        for(int i = 0; i < 3; i ++){
            int idx = 0;
            while(idx < color[i]) {
                nums[cnt++] = i; 
                idx ++;
            }
        } 
    }
};

第一种解法，利用两个指针，一个指向首个头，

表示red的下标，一个指向数组尾，表示blue的下标。 One pass

</pre></p><pre name="code" class="cpp">/**
 *  @author         johnsondu
 *  @time           11:26 14th Oct 2015
 *  @status         Accepted
 *  @strategy       one pass
 *  @problem        Set Matrix Zeroes
 *  @url            https://leetcode.com/problems/set-matrix-zeroes/
 */

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int ridx = 0;
        int bidx = nums.size() - 1;
        int i = 0;
        while(i < bidx + 1)
        {
            if(nums[i] == 0) {
                swap(nums[i], nums[ridx]);
                ridx ++;
                i ++;
                continue;
            }
            if(nums[i] == 2) {
                swap(nums[i], nums[bidx]);
                bidx --;
                continue;
            }
            i ++;
        }
    } 
};