循环不变量:声明的变量在遍历的过程中需要保持定义不变。
设计循环不变量的原则
说明:设计循环不变量的原则是 不重不漏。
以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;
}
}
总结一下:重要点是代码里的注释,
一开始要保证初始化区间为空,和控制循环结束条件。
目的是为了将区间不遗不漏的合并。先移动指针,还是先交换,取决于指针的初始化位置。