1. 题目来源
链接:75. 颜色分类
2. 题目解析
前导题:
采用三指针的思想,将区间分为三段。且考虑边界情况及枚举的选择,本题和前导题都是非常经典的三指针问题!
现在一般三指针来做都是需要划分区间,枚举 i
再在 0~i-1
之间枚举 j
,再在后面枚举 k
,这样的顺序来枚举三个指针。且区间划分是 0~j-1
为第一段区间,j~k-1
为第二段区间,k~n-1
为第三段区间。这个思想及边界情况的考虑和设计都是很巧妙的!!!
体悟算法之美!
- 时间复杂度: O ( n ) O(n) O(n)。
- 空间复杂度: O ( 1 ) O(1) O(1)
代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int n = nums.size();
for (int i = 0, j = 0, k = n - 1; i <= k; ) {
if (nums[i] == 0) swap(nums[i], nums[j]), i ++, j ++ ;
else if (nums[i] == 1) i ++ ;
else if (nums[i] == 2) swap(nums[i], nums[k]), k -- ;
}
}
};
20210718 近期偶然看到面经,是百度二面的笔试题。
算法很不错,思维题。题解区给出了很不错的算法,可以看看官方题解,力荐!
两次遍历:第一次遍历将 0 直接放到首位,并记录放置 0 的个数。第二次遍历,将 1 开始顺序交换。最后剩下的就是 2 全在最后了。
一次遍历:除过上式的方法,官方还给了一种方法,写的有点多,懒得看了。