75. 颜色分类

算法记录

LeetCode 题目：

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

---

---

说明

一、题目

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

二、分析

• 也就是排序只有三种数字的数组，在 n 时间内完成可采用基值交换的方法，每次能将一种数字有序化。

class Solution {
    public void sort(int[] nums,int start, int end, int flag) {
        while(start < end) {
            while(start < end && nums[start] <= flag) start++;
            while(start < end && nums[end] > flag) end--;
            if(start < end) {
                int temp = nums[start];
                nums[start] = nums[end];
                nums[end] = temp;
                start++;
                end--;
            }
        }
    }
    public void sortColors(int[] nums) {
        sort(nums, 0, nums.length - 1, 0);
        sort(nums, 0, nums.length - 1, 1);
    }
}

---

总结

熟悉基值交换排序。