leetcode 75 分类颜色 / sort colours
程序员文章站
2022-05-12 09:31:29
...
题目描述:
这道题应该是一道很经典的题目(以前好像在某司的面试题见过),解题代码也相当具有美感(没做出来参考别人的),设立的三个指针 i,j,k是精髓之处,一定要理解这三个指针的含义,i,j,k分别指向当前已处理好的字符串的最后一个0,1,2。
比如,当前已处理好的字符串是 0 0 0 0 1 1 1 2 2 2 *,那么i,j,k分别指向最后一个0,1,2。m指向*,*号为待处理元素,
i j k m
比如现在*号 的元素为 0 ,那么只需要把原来i,j,k所指元素的后面一个分别赋值成0,1,2 就ok,具体代码如下:
class Solution {
public:
void sortColors(vector<int>& nums) {
int i = -1,j = -1,k = -1;//注意i,j,k的含义
int m;
for(m = 0; m < nums.size();m++){
if(nums[m] == 0){
nums[++k] = 2;
nums[++j] = 1;
nums[++i] = 0;
}
else if(nums[m] == 1){
nums[++k] = 2;
nums[++j] = 1;
}
else {
nums[++k] = 2;
}
}
}
};