【数据结构】关于双指针区间 循环不变量定义

循环不变量：声明的变量在遍历的过程中需要保持定义不变。

设计循环不变量的原则
说明：设计循环不变量的原则是 不重不漏。

以leetcode75作为分析，荷兰国旗问题，将0都移动到1的前面，将2都移动到1的后面。

len 是数组的长度；
变量 zero 是前两个子区间的分界点，一个是闭区间，另一个就必须是开区间；
变量 i 是循环变量，一般设置为开区间，表示 i 之前的元素是遍历过的；
two 是另一个分界线，我设计成闭区间。
如果循环不变量定义如下：

所有在子区间 [0, zero) 的元素都等于 0；
所有在子区间 [zero, i) 的元素都等于 1；
所有在子区间 [two, len - 1] 的元素都等于 2。
于是编码要解决以下三个问题：

变量初始化应该如何定义；
在遍历的时候，是先加减还是先交换；
什么时候循环终止。
处理这三个问题，完全看循环不变量的定义。

• 编码的时候，zero 和 two 初始化的值就应该保证上面的三个子区间全为空；
• 在遍历的过程中，「下标先加减再交换」、还是「先交换再加减」就看初始化的时候变量在哪里；
• 退出循环的条件也看上面定义的循环不变量，在 i == two 成立的时候，上面的三个子区间就正好 不重不漏 地覆盖了整个数组，并且给出的性质成立，题目的任务也就完成了。

下面的代码主要是看 到底是先加减还是先交换。

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

import java.util.Arrays;


public class Solution {

    public void sortColors(int[] nums) {
        int len = nums.length;
        if (len < 2) {
            return;
        }

        // all in [0, zero) = 0
        // all in [zero, i) = 1
        // all in [two, len - 1] = 2
        
        // 循环终止条件是 i == two，那么循环可以继续的条件是 i < two
        // 为了保证初始化的时候 [0, zero) 为空，设置 zero = 0，
        // 所以下面遍历到 0 的时候，先交换，再加
        int zero = 0;

        // 为了保证初始化的时候 [two, len - 1] 为空，设置 two = len
        // 所以下面遍历到 2 的时候，先减，再交换
        int two = len;
        int i = 0;
        // 当 i == two 上面的三个子区间正好覆盖了全部数组
        // 因此，循环可以继续的条件是 i < two
        while (i < two) {
            if (nums[i] == 0) {
                swap(nums, i, zero);
                zero++;
                i++;
            } else if (nums[i] == 1) {
                i++;
            } else {
                two--;
                swap(nums, i, two);
            }
        }
    }

    private void swap(int[] nums, int index1, int index2) {
        int temp = nums[index1];
        nums[index1] = nums[index2];
        nums[index2] = temp;
    }
}

总结一下：重要点是代码里的注释，

一开始要保证初始化区间为空，和控制循环结束条件。

目的是为了将区间不遗不漏的合并。先移动指针，还是先交换，取决于指针的初始化位置。

 

参考：https://leetcode-cn.com/problems/sort-colors/solution/kuai-su-pai-xu-partition-guo-cheng-she-ji-xun-huan/