/*将red,blue颜色的子数组段分别存到nums的头部
和尾部即可。参考自:
https://github.com/soulmachine/leetcode*/
class Solution {
public:
void sortColors(vector<int>& nums) {
int r(0), b(nums.size()-1);
for(int i = 0; i < b+1;){
if(nums[i] == 0){
swap(nums[i++], nums[r++]);
}
else if(nums[i] == 2){
swap(nums[i], nums[b--]);
}
else ++i;
}
}
};
LeetCode之Sort Colors
最新推荐文章于 2019-04-13 19:25:13 发布