Leetcode 075. 颜色分类
地址 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 ;
}
};
作 者: itdef
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
![阿里打赏](https://files.cnblogs.com/files/itdef/ali2.bmp)
![微信打赏](https://files.cnblogs.com/files/itdef/weixin2.bmp)