148.Sort Colors-颜色分类（中等题）

颜色分类

1. 题目

   给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。
   我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

   注意事项
   不能使用代码库中的排序函数来解决这个问题。
   排序需要在原数组中进行。

2. 样例

   给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。

3. 挑战

   一个相当直接的解决方案是使用计数排序扫描2遍的算法。
   首先，迭代数组计算 0,1,2 出现的次数，然后依次用 0,1,2 出现的次数去覆盖数组。
   你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法？

4. 题解

   建立双指针分别指向最后一个0的下一个元素和第一个2的前一个元素，遍历数组，找到0则与left指针交换，找到2则与right指针交换。

class Solution {
    /**
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */
    public void sortColors(int[] nums) {
        int left = 0;
        int right = nums.length-1;
        int i=0;
        while (i <= right)
        {
            if (nums[i] == 0)
            {
                swap(nums,i,left);
                left++;
                i++;
            }
            else if (nums[i] == 1)
            {
                i++;
            }
            else
            {
                swap(nums,i,right);
                right--;
            }
        }
    }

    private void swap(int[] a, int i, int j) 
    {
        int tmp = a[i];
        a[i] = a[j];
        a[j] = tmp;
    }
}

Last Update 2016.10.17