class Solution {
public:
void sortColors(vector<int>& nums) {
int l = 0, r = nums.size() - 1;
for(int i = 0; i <= r; ++i){
if(nums[i] == 0){
swap(nums[i], nums[l]);
++l; //这里不用--i是因为换过来的nums[l]肯定是1
}else if(nums[i] == 2){
swap(nums[i], nums[r]);
--r;
--i; //换过来的nums[r]可能是2,还得重新判断一次
}
}
}
};
LeetCode 75. 颜色分类
最新推荐文章于 2024-10-06 14:10:45 发布