标签：p0 p1 颜色 nums ++ 分类 int swap leetcode

简单粗暴：直接对数组进行sort()排序

思路1：(单指针，两次遍历)
1）对元素为0的交换到前面
2）对元素为1的交换到0后面

代码1：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p = 0;
        //交换0
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 0) {
                swap(nums[i], nums[p]);
                p++;
            }
        }
        //交换1
        for (int i = p; i < nums.size(); i++) {
            if (nums[i] == 1) {
                swap(nums[i], nums[p]);
                p++;
            }
        }
    }
};

思路2：（双指针，一次遍历）
1）对元素为1的交换到前面
2）对元素为0的交换到前面p0，同时因为将1换出，需要将值放入p1中

代码2：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0 = 0, p1 = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 1) {
                swap(nums[i], nums[p1]);
                p1++;
            } 
            else if (nums[i] == 0) {
                swap(nums[i], nums[p0]);
                if (p0 < p1)//将num[i]放入nums[p0],当前num[i]为1,放入nums[p1]
                    swap(nums[i], nums[p1]);
                p0++;
                p1++;
            }
        }
    }
};

思路3：
1）对元素为2的交换到后面（注意头元素和尾元素都为2的情况）
2）对元素为0的交换到前面

代码3：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0 = 0, p2 = nums.size() - 1;
        for (int i = 0; i <= p2; i++) {
            //处理num[i]与num[p2]都为2的情况
            while (i <= p2 && nums[i] == 2) {
                swap(nums[i], nums[p2]);
                p2--;
            }
            //将0交换到前面
            if (nums[i] == 0) {
                swap(nums[i], nums[p0]);
                p0++;
            }
        }
    }
};

标签：p0,p1,颜色,nums,++,分类,int,swap,leetcode
来源： https://blog.csdn.net/Dedication_/article/details/115533495

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。