leetcode hot100 之颜色分类

最新推荐文章于 2024-12-29 23:20:44 发布

原创最新推荐文章于 2024-12-29 23:20:44 发布 · 857 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

LeetCode 专栏收录该内容

143 篇文章

订阅专栏

本文介绍了一种解决LeetCode上的颜色排序问题的方法，即双指针法。给定一个包含红、白、蓝三种颜色的数组，原地将其排序。通过维护两个指针start和end，将0移动到前面，1移动到中间，2移动到后面，实现O(n)的时间复杂度和O(1)的空间复杂度。代码中展示了具体的实现细节。

题目

假定有红、白、蓝三个颜色，分别用 0、1、2 来表示。原地对他们进行排序。

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

原题链接：https://leetcode-cn.com/problems/sort-colors/

思路

双指针法：用 start 和 end 指向序列1 的起始和末尾，具体的是 [start, end)。如果 start < end，则说明这个区间不是一个空序列。遍历数组，如果遇到 0，则把 0 塞到 start 处，如果区间不是空序列，需要把置换出来的 1 塞到 end 处；如果遇到 1，则把 1 塞到 end 处。直到遍历完数组为止。
统计法：直接遍历一遍，统计有多少个0，多少个1，多少个2。然后重写一次数组。

复杂度分析
- 时间复杂度 O(n)
- 空间复杂度 O(1)

代码

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