LeetCode琅琊榜-第一层(荷兰国旗问题)

LeetCode_75.颜色分类

难度:中等

关注博主,持续输出优质算法内容

题目

---

传统代码的思路与实现🐼

class Solution {
    public void sortColors(int[] nums) {
        var red = 0;
        var white = 0;
        var blue = 0;
        for (var item : nums) {
            if (item == 0) {
                red++;
            }else if (item == 1) {
                white++;
            }else {
                blue++;
            }
        }
        for (int i = 0;i < nums.length;i++) {
            if (i < red) {
                nums[i] = 0;
            }else if (i < red + white) {
                nums[i] = 1;
            }else {
                nums[i] = 2;
            }
        }
    }
}

• 1.思路🦁

  • 由题目可知,数组里面的元素有且仅有三个元素,分别是[0,1,2]
  • 根据示例可知,无非是将这三个数字进行排序罢了
  • 我们不妨把这三个数字的个数分别统计起来,然后再将它们分别赋值给数组中对应的下标

• 2.代码分析🦁

  • 如代码所示,第一个增强for循环用于对这三个数字分别进行统计,相应的我们要设置三个对应的变量用于统计对应的个数
  • 第二个for循环用于给数组对应的位置赋值

---

不足之处及提升的方法🐼 

不足:😀

• 该算法循环遍历了该数组两次,太多了,我不满意

提升的方法:😀

• 采用快速排序的思想来完成这道问题,即这道题对应的荷兰国旗问题算法的思想

---

荷兰国旗问题的思路分析🐼

• 数组中只可能存在三个不同的元素,我们不妨把他们分为三组
  • [0,p0),这一组只可以放0
  • [p0,i),这一组只可以放1
  • [p2,nums.lengh-1]这一组只能放2
  • 要点:
    • 0和nums.length-1是闭区间的原因是这两个位置是可以肯定的,因为第一个区间一定从0开始,最后一个区间一定从nums.length-1结束
    • 同理得,第二个区间一定是从p0开始,其他都是不确定的,所以用开区间
• 我们分别定义上面的三个变量i,p0,p1,然后得出一下的定义
  • p0表示的是第一个区间的最后一个元素的后一个元素
  • i表示的是第二区间的最后一个元素的后一个元素
  • p2表示的是第三个区间的第一个元素的前一个元素
• 注意:我们要严格遵循变量定义不变的规则,我们不能写着写着程序这些变量的所对应的意义发生了改变,我更愿意称之为"变量定义不变性"
• 注意:这三个区间一开始应该是空的,因为一开始我们不知道谁应该有多少,有没有,所以应该是空的

---

荷兰国旗问题的代码实现与分析🐼

class Solution {
    public void sortColors(int[] nums) {
        var p0 = 0;
        var p1 = nums.length - 1;
        int i = 0;
        while (i <= p1) {
            if (nums[i] == 0) {
                swap(nums,i,p0);
                p0++;
                i++;
            }else if (nums[i] == 1) {
                i++;
            }else {
                swap(nums,i,p1);
                p1--;
            }
        } 
    }

    public void swap(int[] nums,int n1,int n2) {
        var temp = nums[n1];
        nums[n1] = nums[n2];
        nums[n2] = temp;
    }
}

代码分析🐨:

1. p0和i赋值为0,p1赋值为nums.length-1即他们三个对应的区间分别是[0,0) [0,0) (length-1,length-1]
   1. 可以清晰的得出他们三个区间一开始都是空区间
2. 循环讲述
   1. 布尔表达式为i <= p1的原因
      1. 我们不妨看看定义,我们这三个区间最终组合起来应该是整个数组,我们不妨得到[0,p0) + [p0,i) + (p1,nums.length-1]
      2. 当我们的i = p1的时候刚好遍历到最后一个元素,即当i=p1的时候,就可以组成一个完整的数组
   2. 如果我们发现,i指向的元素是等于0的话,我们应该放在第一个区间中,由定义可知,p0是最后一个元素的后一个元素,所以i要与p0调换
      1. 调换之后,为了保持变量定义不变性,我们需要将p0++,因为p0++了,对应的i也要++,如果i不++,会导致第二个区间的i的定义发生了变化
   3. 如果我们发现i指向的元素等于1的话,由于变量定义不变性,i需要++
   4. 同理得
      1. 注意的是:如果放到第三个区间,i的定义并没有发生改变,所以不需要进行改变

---

结论🐼

        题目的解法是以快速排序的基础上建立起来的,然而,在我的观点看来,这可以独立成一个区间问题,以后凡是遇到区间问题都可以借鉴该算法,我来总结该算法的核心掌握内容

        1.变量定义不变性

        2.区间的划分