三色排序

来源:互联网 发布:h3c mac认证 编辑:程序博客网 时间:2024/03/29 07:59

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

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

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

思路:与快排划分过程相似
class ThreeColor {public:    vector<int> sortThreeColor(vector<int> A, int n) {         // 利用快排思想        int low=0,high = n-1;        for(int i = 0;i<=high;i++){            if(A[i]==0) {                swap(A[i],A[low]);                low ++;            }else if(A[i]==2){                swap(A[i],A[high]);                high--;                i--;//交换过来的A[high]未知,需要i从新比较一次            }        }        return A;    }};