【leetCode】leetCode75 sortColors java解法

题目地址：https://leetcode.com/problems/sort-colors/

方法一、最容易想到，计数排序，统计0,1,2出现的个数，再逐一打印。需要遍历数组两次。

方法二、类似三路快排。定义两个指针，[0,Zero]全是0，[Two,arr.length]全是2，中间是1和未判断部分。

（方法二做时有一个细节问题，就是指针移动的问题。因为比较完不是2以后，还不能移动指针，还需要比较是不是0，以后再移动指针。如果是1 的话就直接移动指针。）

package arrays;
//1.计数排序
public class sortColors75 {
    //计数排序
    private static int[] sortColors1 (int[] arr) throws Exception {
        int[] count = new int[3]; 
        for(int i=0;i<arr.length;i++){
            if(arr[i]<0||arr[i]>2){
                throw new Exception("输入范围不合理");
            }
            count[arr[i]]++;
        }
            
        int index=0;
        for(int j=0;j<count[0];j++) {
            arr[index++]=0;
        }
        for(int j=0;j<count[1];j++) {
            arr[index++]=1;
        }            
        for(int j=0;j<count[2];j++) {
            arr[index++]=2;
        }                
        return arr;
        
    }
//2.指针    
    private static int[] sortColors2 (int[] arr) {
        int Zero =-1;
        int Two =arr.length;
        int index=0;
        while (index<Two){            
            if(arr[index]==2){
                Two--;
                swapInt(Two,index,arr);
            }            
            else if (arr[index]==0){
                Zero++;
                swapInt(Zero,index,arr);
                index++;
            }    
            else 
                 index++;
        }
        return arr;
        
    }
    private static void swapInt(int i,int j,int[] arr){
        int tmp= arr[i];
        arr[i]=arr[j];
        arr[j]=tmp;
    }
    public static void main(String[] args) throws Exception {    
    int [] arr = {1,0,1,2,1,2};
    arr = sortColors2(arr);
    for(int i =0 ; i<arr.length;i++) {
        System.out.print(arr[i]+" ");
    }

}
}