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啦