LeetCode之Sort Colors

来源:互联网 发布:淘宝商城玩具大全 编辑:程序博客网 时间:2024/06/08 20:48
/*将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;        }    }};

0 0