Leetcode 347. Sort Colors

本题就是荷兰国旗问题, 思路较为简单, 和快排的思路较为类似, 在数组的左边区间放 0, 初始区间的位置为 -1; 数组的右边区间放 2, 初始区间的位置为 nums.length; 然后遍历该数组, 如果该数为 0, 那么就和左边区间的下一个元素位置交换, 如果为 2, 那么就和右边区间的上一个元素位置交换. 但是要注意的是, 左边的元素已经遍历过了, 所以交换了之后也可以直接跳过该位置; 但是右边交换过来的没有遍历过, 所以不能跳过该位置.

还有一点就是 while 的条件是 index < right, 不需要遍历到数组的最右边, 因为 right 代表的右边区间都是 2 , 所以不需要遍历了. 

class Solution {
    public void sortColors(int[] nums) {
        int left = -1, right = nums.length;
        int index = 0;
        while(index < right) {
            if(nums[index] == 0)
                swap(nums, ++left, index++);
            else if(nums[index] == 2)
                swap(nums, --right, index);
            else
                index++;
        }
        return;
    }
    public void swap(int[] arr, int index1, int index2) {
        int temp = arr[index1];
        arr[index1] = arr[index2];
        arr[index2] = temp;
    }
}