python颜色代码有哪些_Python测试篇：LeetCode 75. 颜色分类

上班第一天，有没有感觉些许困意？来道LeetCode测试题提提神吧！今天我们看看来自力扣的75题，关于颜色分类。

题目

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:

不能使用代码库中的排序函数来解决这道题。

示例:

进阶：

1、一个直观的解决方案是使用计数排序的两趟扫描算法。

首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。

2、你能想出一个仅使用常数空间的一趟扫描算法吗？

解题思路

思路：

这道题的本质还是一道排序的题，题目中给出提示说可以用计数排序，需要遍历数组两遍，那么先来看这种方法，因为数组中只有三个不同的元素，所以实现起来很容易。

- 首先遍历一遍原数组，分别记录0,1,2的个数

- 然后更新原数组，按个数分别赋上0，1，2

计数排序

我们先统计总共有多少个0、1和2，然后再将这些值放回就可以了，这实际上是计数排序的原理。

先创建长度为 3 的数组 count_array，用来迭代统计0、1、2三个元素的个数；

遍历，根据 count_array 记录的个数，按照 0、1、2 的顺序重写数组。

我们通过分析发现这个算法的时间复杂度是O(n)级别的，而空间复杂度和我们统计的数的个数有关O(3)。并且我们这个解法中扫描了两遍数组，我们有没有只扫描一遍的做法呢？

双指针

题目中还要让只遍历一次数组来求解，那么我们需要用双指针来做，分别从原数组的首尾往中心移动。

- 定义red指针指向开头位置，blue指针指向末尾位置

- 从头开始遍历原数组，如果遇到0，则交换该值和red指针指向的值，并将red指针后移一位。若遇到2，则交换该值和blue指针指向的值，并将blue指针前移一位。若遇到1，则继续遍历。

详细步骤如下：

1、初始化双指针 p1、p2。其中 p1 指向数组头部，p2 指向数组尾部。

2、定义游标 cur 指向数组头部，从左往右遍历数组：

当 nums[cur] == 0 时，交换 cur 对应的元素与 p1 对应的元素，同时将 cur、p1 往后移动一个位置。这里可能存在这样的情况，nums[cur] 等于 0，但 cur 与 p1 重合，这里直接将 cur、p1 往后移动一个位置；

当 nums[cur] == 2 时，交换 cur 对应的元素与 p2 对应的元素，仅将 p2 往前移动一个位置。

若不是上面两种情况(也就是 cur 指向 1 时)，仅将 cur 往后移动一个位置。

3、由于遍历从左往右，p2 由右往前移动。所以当 cur 大于 p2 时，停止遍历。

下面是图示：

代码如下：

复杂度分析

时间复杂度：O ( n ) O(n)O(n)，n nn 是数组 n u m s numsnums 的长度。

空间复杂度：O ( 1 ) O(1)O(1)，仅使用了常数个变量。