Leetcode 75. Sort Colors
原创
©著作权归作者所有:来自51CTO博客作者Nemaleswang的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目链接:75. Sort Colors
题目大意:给定一个数组,里面只包含0,1,2三个数,要求只做一次线性扫描,将数组变得有序
题目思路:我们只有0,1,2三种不同的数,我们只需要考虑把0放在前面,2放在后面即可,类比快排,我们可以考虑这样的情况,我们定义三个指针,l,r,mid,l之前的值我们置为0,r之后的值我们置为2,mid我们用做扫值,碰到0就往前面扔,碰到2,就往后面扔,碰到2的时候需要注意,置换后的值可能还需要往前扔,所以,mid值不能变,但是碰到0的时候不需要暂停,因为前面的值只可能是0和1,讲的有点乱,具体看代码。
时间复杂度&&空间复杂度:O(n)(只扫一遍)&&O(1)
class Solution {
public:
void sortColors(vector<int>& nums) {
int l = 0,mid = 0,r = nums.size()-1;
while(mid <= r){
if(nums[mid] == 0){
swap(nums[mid++],nums[l++]);
}
else if(nums[mid] == 2){
swap(nums[mid],nums[r--]);
}
else mid++;
}
}
};