三色排序 -- 算法小结

题目描述:有一个只由0，1，2三种元素构成的整数数组，请使用交换、原地排序而不是使用计数进行排序。

给定一个只含0，1，2的整数数组A及它的大小，请返回排序后的数组。保证数组大小小于等于500。

测试样例：
[0,1,1,0,2,2],6
返回：[0,0,1,1,2,2]

import java.util.*;public class ThreeColor {    public int[] sortThreeColor(int[] A, int n) {        // write code here        int left = 0;        int right = n-1;        for(int i=0;i<n;i++){            if(right<i){                break;            }            else if(A[i] == 0 &&left<=right){                change(A,left,i);                left++;            }             else if(A[i] == 2 &&left<=right){                change(A,right,i);                right--;                i--;            }        }        return A;    }     public void change(int[] A,int a,int b){            if(a!=b){                int tem =A[a];                A[a] = A[b];                A[b] = tem;            }        }}