只包含特定元素的数组排序

//多指针
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param array int整型vector 
     * @return int整型vector
     */
    vector<int> sort(vector<int>& array) {
        // write code here
        int i=0, p0=0, p2=array.size()-1;
        while(i<=p2){
            while(array[p0]==0) p0++;
            if(i<p0)i=p0;
            while(array[p2]==2) p2--;
            if(i>p2) break;
            if(array[i]==0){
                swap(array[i],array[p0]);
                p0++;
            }else if(array[i]==2){
                swap(array[i],array[p2]);
                p2--;
            }else if(array[i]==1){
                i++;
            }
        }

        return array;
    }
};

import java.util.*;
 
 
public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param array int整型一维数组
     * @return int整型一维数组
     */
    public int[] sort (int[] array) {
        // write code here
        int[] arr = new int[array.length];
        int index1 = 0;
        int index2 = array.length - 1;
        for (int i = 0; i < array.length; i++) {
            arr[i] = 1;
        }
 
        for (int i = 0; i < array.length; i++) {
            if (array[i] == 0) {
                arr[index1++] = 0;
            }else if (array[i] == 2) {
                arr[index2--] = 2;
            }
        }
 
        return arr;
    }
}