No.217 - LeetCode[75] Sort Colors - 数组中三类元素划分

原创于 2020-09-04 17:55:20 发布 · 175 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用三个游标（L，mid，R）的高效算法，用于解决LeetCode上的“75.颜色分类”问题。该算法通过主循环遍历mid游标，根据不同条件交换元素位置，实现0、1、2三种颜色的快速排序。

核心思想，三个游标，L，mid，R
以mid游标为主循环

/*
 * @lc app=leetcode id=75 lang=cpp
 *
 * [75] Sort Colors
 */

// @lc code=start
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int N = nums.size();
        int L = 0;
        int R = N-1;
        int mid = 0;
        while(mid <= R){
            if(nums[mid] == 0){
                swap(nums[mid],nums[L]);
                L++;
                mid++;
                continue;
            }
            if(nums[mid] == 1){
                mid++;
                continue;
            }
            if(nums[mid] == 2){
                swap(nums[mid],nums[R]);
                R--;
                continue;
            }
        }
    }
};
// @lc code=end