标签：075 示例 nums Leetcode x2 x1 分类 find 输入

地址 https://leetcode-cn.com/problems/sort-colors/

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，
使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例 1：
输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：
输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：
输入：nums = [0]
输出：[0]

示例 4：
输入：nums = [1]
输出：[1]

解答
首先朴素想法是排序 就能得到题目要求，但是复杂度是O(nlogn);

class Solution {
public:
    void sortColors(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        return ;
    }
};

可以想到 整个数组有两个边界 x1 和 x2

x1左边全部是0 x1右边全部是1
x2左边全是1 x2右边全是2

扫描两次即可，每次都是从头到尾扫描，时间复杂度也是O(n)

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int first = 0; int find =0;
        //首先保证0 全部归位
        for(find  =0;find<nums.size();find++){
            if(nums[find]==0){
                swap(nums[first],nums[find]);
                first++;
            }
        }
        //0全部归位后， 查找1
        for(find = first;find<nums.size();find++){
            if(nums[find]==1){
                swap(nums[first],nums[find]);
                first++;
            }
        } 
        return ;
    }
};

我的视频题解空间

标签：075,示例,nums,Leetcode,x2,x1,分类,find,输入
来源： https://www.cnblogs.com/itdef/p/15677188.html

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