颜色分类

Posted 2022-08-23 zcxhaha

题目链接：

涉及知识：数组、双指针

问题描述：

? 该问题又被称为荷兰国旗问题，大概意思为给每一个数字指定一种颜色，并按荷兰国旗颜色的顺序进行调整。

算法思想：

? 设置三个指针，初始状态 low 指针指向最低位， high 指针指向高位，并设置 cur 指针指向当前的元素。cur 指针从头开始遍历。

• 初始化 0 的最右边界：low = 0。

• 初始化 2 的最左边界 ：high = n - 1。

• 初始化当前考虑的元素序号 ：cur = 0.
  • While cur <= high :
  • 若 nums[cur] = 0 ：交换第 cur 个和第 low 个元素，并将指针都向右移。
  • 若 nums[cur] = 2 ：交换第 cur 个和第 high 个元素，并将 high 指针左移 。
  • 若 nums[cur] = 1 ：将指针 cur 右移。

为什么和低位交换之后，两个指针均向右移，而和高位交换之后只将 high 指针向左移动？

? 因为和低位交换之后，cur 所指向的元素为 1（原先的 low 指针指向的为 1，当然刚开始的时候可能为 0，但是没有影响），而 high 指针原先指向的元素可能为 2，交换之后 cur 指向的元素可能为 2，需要在和 high 指向的元素进行交换，因此此时不可以将 cur 向右移动。

代码：

class Solution 
    public void sortColors(int[] nums) 
        int low = 0;
        int high = nums.length - 1;
        int cur = 0;

        while(cur <= high)
            if(nums[cur] == 0)
                int tmp = nums[cur];
                nums[cur] = nums[low];
                nums[low] = tmp;
                low++;
                cur++;
            else if(nums[cur] == 2) 
                int tmp = nums[cur];
                nums[cur] = nums[high];
                nums[high] = tmp;
                high--;
             else 
                cur++;
            
        

        return;