算法--三色排序练习题

import java.util.*;

public class ThreeColor {

   public int[] sortThreeColor(int[] A, int n) {

        int zeroAreaIndex = -1;//0区域当前最后位置

        int twoAreaIndex = n;//2区域当前最前位置

        for(int i=0; i<A.length; ){

            if(A[i] == 1) i++;

            else if(A[i] == 0){

                swap(A, ++zeroAreaIndex, i);//当前位置为0，和0区域的下一个位置进行交换(0 0 0 1 0 2 0 1)

                i++;

            }

            else{//当前位置为2，那么和2区域的前一个位置进行交换(0 0 0 1 2 1 1 1 0 2)

                swap(A, --twoAreaIndex, i);

                if(i == twoAreaIndex) break;

                /**

                 * 此时从2区域前方交换到了当前i位置，不知道2区域前方这个数是1还是2还是0？

                 * 如果是1的话，i++,继续下一次循环;

                 * 如果是0或者2的话，i保持不变

                 */

                if(A[i] == 1) i++;

                else continue;

            }

            if(i == twoAreaIndex) break;

        }

        return A;

    }

    public void swap(int[] arr, int m, int n) {

        int temp = arr[m];

        arr[m] = arr[n];

        arr[n] = temp;

    }

}