用Java解决一个颜色分类问题

Question

颜色分类
给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地(https://baike.baidu.com/item/%E5%8E%9F%E5%9C%B0%E7%AE%97%E6%B3%95)%E5%AF%B9%E5%AE%83%E4%BB%AC%E8%BF%9B%E8%A1%8C%E6%8E%92%E5%BA%8F%EF%BC%8C%E4%BD%BF%E5%BE%97%E7%9B%B8%E5%90%8C%E9%A2%9C%E8%89%B2%E7%9A%84%E5%85%83%E7%B4%A0%E7%9B%B8%E9%82%BB%EF%BC%8C%E5%B9%B6%E6%8C%89%E7%85%A7%E7%BA%A2%E8%89%B2%E3%80%81%E7%99%BD%E8%89%B2%E3%80%81%E8%93%9D%E8%89%B2%E9%A1%BA%E5%BA%8F%E6%8E%92%E5%88%97%E3%80%82

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

示例 1：

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

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

输入：nums = [0]
输出：[0]
示例 4：

输入：nums = [1]
输出：[1]

提示：

n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

Answer 1

class Solution {
    public void sortColors(int[] nums) {
        int low = 0, high = nums.length - 1;
        int i = 0;
        while (i <= high) {
            if (nums[i] == 0) {
                int tmp = nums[i];
                nums[i] = nums[low];
                nums[low] = tmp;
                ++low;
                ++i;
            } else if (nums[i] == 1) {
                ++i;
            } else if (i <= high && nums[i] == 2) {
                int tmp = nums[i];
                nums[i] = nums[high];
                nums[high] = tmp;
                --high;
            }
        }
    }
}