【leetcode】Sort Colors

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int count[3] ={0, 0, 0};
        int element = 0;
        for(int i = 0; i < nums.size(); i++){
            count[nums[i]]++;
        }
        for(int i = 0; i < 3; i++){
            for(int j = 0; j < count[i]; j++){
                nums[element++] = i;
            }
        }
    }
};

以上是AC代码

思路很简单

先统计每个颜色有多少个

然后再按照0,1,2的顺序写入nums数组就Ok啦