LintCode——颜色分类

颜色分类：给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

　　　　　我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

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

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

方法一：常规解决方案(1、迭代数组计算 0,1,2 出现的次数；2、然后依次用 0,1,2 出现的次数去覆盖数组)

 class Solution {
     /**
      * @param nums: A list of integer which is 0, 1 or 2
      * @return: nothing
      */
     public void sortColors(int[] nums) {
         if(null == nums || nums.length <= 1) return;
         int Num0 = 0, Num1 = 0, Num2 = 0;
         for(int i = 0; i < nums.length; i++) {
             switch(nums[i]) {
                 case 0:
                     Num0++;
                     break;
                 case 1:
                     Num1++;
                     break;
                 case 2:
                     Num2++;
                     break;
                 default:
                     return;
             }
         }
         for(int i = 0; i < nums.length; i++) {
             if(i < Num0) {
                 nums[i] = 0;
             }else if(i >= Num0 && i < Num0+Num1) {
                 nums[i] = 1;
             }else {
                 nums[i] = 2;
             }
         }
     }
 }

方法二：题目要求(仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法)

 public class Solution {
     /**
      * @param nums: A list of integer which is 0, 1 or 2
      * @return: nothing
      */
     public void sortColors(int[] nums) {
         int i = 0,I = 0;
         int r = nums.length-1;
         int temp;
         //System.out.printf("r=%d",r);

         for(i=0;i<nums.length-1&&i<r+1;){
             if(nums[i]==0){
                 if(i==I){
                     i++;
                 }
                 else{
                     temp=nums[i];
                     nums[i]=nums[I];
                     nums[I]=temp;
                 }
                 I++;
             }
             else if(nums[i]==2){
                 temp=nums[i];
                 nums[i]=nums[r];
                 nums[r]=temp;
                 r--;
             }
             else{
                 i++;
             }
         }
     }
 }