Leetcode 75. 颜色分类

1、问题分析

题目链接：https://leetcode-cn.com/problems/sort-colors/
  本质上就是一个排序问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

2、问题解决

  笔者以C++方式解决。

#include "iostream"

using namespace std;

#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"

class Solution {
public:
    void sortColors(vector<int> &nums) {
        // 如果数组为空，直接返回即可
        if (nums.empty()) {
            return;
        }
        // 使用自定义快排
        quickSOrt(nums, 0, nums.size() - 1);
    }

    void quickSOrt(vector<int> &nums, int left, int right) {
        if (left < right) {
            int pos = partition(nums, left, right);
            quickSOrt(nums, left, pos - 1);
            quickSOrt(nums, pos + 1, right);
        }
    }

    int partition(vector<int> &nums, int left, int right) {
        int temp = nums[left];
        while (left < right) {
            while (left < right && nums[right] > temp) {
                right--;
            }

            nums[left] = nums[right];
            while (left < right && nums[left] <= nums[right]) {
                left++;
            }
            nums[right] = nums[left];
        }
        nums[left] = temp;
        return left;

    }
};

int main() {
    vector<int> nums = {2, 0, 2, 1, 1, 0};
    Solution *pSolution = new Solution;
    pSolution->sortColors(nums);

    for (int i = 0; i < nums.size(); ++i) {
        cout << nums[i] << " ";
    }

    cout << endl;

    system("pause");
    return 0;
}

运行结果

有点菜，有时间再优化一下。

3、总结

  难得有时间刷一波LeetCode, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！
哪怕只有一个人从我的博客受益，我也知足了。
点个赞再走呗！欢迎留言哦！