【双指针】Leetcode --- NO.75 颜色分类（Java）

题目描述

题目分析

• 这道题给定得数组中只有 0 1 2 这三个数字，所以唯一要做的就是排序
• 最直接的方法就是直接双重遍历排序
• 可以使用多指针的方法，尽量一次遍历就完成排序
• 可以使用两个个指针，分别指向 0 1 2 三个数字的交界处
• 如果当前值是 0 就往头部放，如果是 2 就往尾部放，这样中间的值自然就是 1 了
• 再用这两个指针和当前遍历的值进行判断，直到数组所有元素全部遍历一遍为止

解法分析

• 使用 while 循环，直到当前遍历的指针与指向 1 和 2的交界处 的指针相遇，表示已经排序完成
• 如果当前值为0，就让当前值和指向0 和 1的交界处 的值交换，继续遍历下一个
• 如果当前值为 2，就让当前值和指向1 和 2的交界处 的值交换，这里要注意，因为不能保证交换的值就是1，所以需要继续判断一次，也就是当前指针不前进

代码

class Solution {
    public void sortColors(int[] nums) {
        // 定义两个指针，分别表示 0 1 2 三个数字的交界处
        // 初始为头和尾
        int zero = 0;
        int two = nums.length-1;
        // 当前遍历指针
        int cur = 0;
        // 交换值所用的临时变量
        int temp;
        // 循环遍历，直到当前遍历的指针和 2 的指针相遇
        while(cur <= two){
            if(nums[cur] == 0){
                // 如果值为 0，就交换放到前面
                // 因为前面的数字肯定是 0 或 1 ，所以交换过后 1 就到中间了
                // 因此可以继续判断下一个
                temp = nums[zero];
                nums[zero++] = nums[cur];
                nums[cur++] = temp;
            } else if(nums[cur] == 2){
                // 如果值为 2 ，就和后面的交换
                // 但是后面的值有可能是 0 或 1 ，所以交换过后仍需要继续判断一次
                // 也就是不前进
                temp = nums[cur];
                nums[cur] = nums[two];
                nums[two--] = temp;
            } else {
                // 如果是 1，就让他继续在中间待着
                cur++;
            }
        }
    }
}