LintCode_颜色分类

问题描述：

算法描述：

根据上面的说明，我设计了两种算法，第一种遍历两次，第二种只遍历一次

方法一：

    public static void sortColors1(int[] nums) {
        
        if(nums!=null&&nums.length!=0){
            
            int size=nums.length;
            int[] count=new int[3];
            int pos=0;
            
            //统计0,1,2的个数
            for(int i=0;i<size;i++){
                
                if(nums[i]==0){
                    count[0]++;
                }else if(nums[i]==1){
                    count[1]++;
                }else if(nums[i]==2){
                    count[2]++;
                }
            }
            
            //再遍历一次数组，改变nums
            for(int j=0;j<count.length;j++){
                while(count[j]!=0){
                    nums[pos]=j;
                    pos++;
                    count[j]--;
                }
            }
        }

//        for(int k:nums){
//            System.out.print(k+" ");
//        }
    }

方法二：

    public static void sortColors(int[] nums) {
        if(nums!=null&&nums.length!=0){
            
            //记录0和2的位置
            int pos=0;
            //nums的长度
            int size=nums.length;
            //记录已检测到的2的个数，这个在将2插入到最后有很大作用，以免陷入死循环
            int count2=0;
            //起到一个标记作用，主要是在将待插的2后面有0或者2，这样当后面的数往前移一位了，但i++，这样可能跳过了移位后待插处的0或者2；
            int flag2=0;
            for(int i=0;i<size-count2;i++){
                if(nums[i]==0){
                    pos=i;
                    while(pos>0){
                        //遇到0，则把0前面的数往后移，然后将0插入到数组第一个
                        nums[pos]=nums[pos-1];
                        pos--;
                    }
                    nums[pos]=0;
                }else if(nums[i]==2){
                    pos=i;
                    count2++;
                    flag2=2;
                    while(pos<size-1){
                      //遇到2，则把2后面的数往后移，然后将2插入到数组最后一个
                        nums[pos]=nums[pos+1];
                        pos++;
                    }
                    nums[pos]=2;
                }
                
                if(flag2==2&&(nums[i]==0||nums[i]==2)&&i<size-count2){
                    i--;
                    flag2=0;
                }
            }
        }
    }