欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leetcode 75 分类颜色 / sort colours

程序员文章站 2022-05-12 09:31:29
...

题目描述:

leetcode 75 分类颜色 / sort colours

   这道题应该是一道很经典的题目(以前好像在某司的面试题见过),解题代码也相当具有美感(没做出来参考别人的),设立的三个指针 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;
            }
        }
    }
};