LeetCode之Sort Colors

/*将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;
}
}
};