LeetCode------Sort Colors

1、计数排序

    //计数排序：两边扫描，第一遍计数，第二遍给数组重新赋值
    public void sortColors(int nums[]) {
        int counts[] = new int[3];
        //计数
        for(int i = 0;i<nums.length;i++) {
            counts[nums[i]]++;
        }
        //给数组重新赋值
        for(int i= 0,index = 0;i<3;i++) {
            for(int j = 0;j<counts[i];j++) {
                nums[index++] = i;
            }
        }
    }

2、双指针

只遍历一次数组来求解，那么我需要用双指针来做，分别从原数组的首尾往中心移动。

• 定义red指针指向开头位置，blue指针指向末尾位置

• 从头开始遍历原数组，如果遇到0，则交换该值和red指针指向的值，并将red指针后移一位。若遇到2，则交换该值和blue指针指向的值，并将blue指针前移一位。若遇到1，则继续遍历。

    //双指针：一个是red的index，一个是blue的index
    public void sortColors2(int nums[]) {
        // 一个是red的index，一个是blue的index，两边往中间走
        int red = 0, blue = nums.length - 1;
        for (int i = 0; i < blue + 1;) {
            if (nums[i] == 0)
                swap (nums, i++, red++);
            else if (nums[i] == 2)
                swap(nums, i, blue--);
            else
                i++;
        }
    } 
    //交换
    private static void swap(int[] A, int i, int j) {
        int tmp = A[i];
        A[i] = A[j];
        A[j] = tmp;
    }