- 题目描述
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色
的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
- 题解
排序
- 代码
class Solution {
public:
void sortColors(vector<int>& nums) {
int l=0,h=nums.size()-1;
for (int i=l;i<=h;i++){
if (nums[i]==0){
swap(nums[i],nums[l]);
l++;
}else if (nums[i]==2){
swap(nums[i],nums[h]);
h--;
i--;
}
}
}
};