颜色分类

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

示例

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

思路
此题可以使用双指针法解决,设置下标i,j,使得i所指的元素之前的位置均为0,j所指的元素之后的位置均为2,使得i,j之间的元素均为1。

代码:时间复杂度o(n),空间复杂度o(1)

void sortColors(vector<int>& nums) {
		//i初始指向第一个元素,j指向最后一个元素
		//k用来遍历
        int i = 0, j = nums.size()-1, k = 0;
        while(k <= j)
        {
        	//如果k所指向的元素为0,就将其置换到最前面
            if(nums[k] == 0){swap(nums[k],nums[i]);i++;k++;}
            //如果k所指向的元素为2,就将其置换到最后面
            else if(nums[k] == 2) {swap(nums[k],nums[j]); j--;}
            //如果k所指向的元素为1,就继续向后遍历
            else k++;
        }
    }
发布了19 篇原创文章 · 获赞 0 · 访问量 117
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 数字20 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览