LeetCode算法题75:颜色分类解析

给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶:

  • 一个直观的解决方案是使用计数排序的两趟扫描算法。
  • 首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
  • 你能想出一个仅使用常数空间的一趟扫描算法吗?

这个题提示的思路就是直接两趟扫描计数,这个比较简单,使用C++编写,另外一种可以利用只有三种颜色,红色开头,蓝色结尾,那么只需要设置两个指针,红指针指向头和蓝指针指向尾,如果碰到0就与红指针指向的数字交换,然后红指针加1,如果碰到2,那么与蓝色指针指向的数字交换,然后蓝色指针减1,需要注意的是,由于遍历都是从前往后,所以防止出现将0换到前面的情况,与蓝指针交换完之后需要退一步,否则就会出现跳过遍历的情况,另外,遍历时,直接遍历蓝色指针个数即可,这个数字是不停变换的,当当前步大于蓝色指针时,说明已经是排好序了。

C++源代码:

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        int count[3] = {0};
        int k = 0;
        for (int i=0;i<n;i++)
            count[nums[i]]++;
        for (int i=0;i<3;i++)
            for(int j=0;j<count[i];j++)
                nums[k++] = i;
    }
};

python3源代码:

class Solution:
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        r = 0
        b = n-1
        i = 0
        while i <= b:
            if nums[i] == 0:
                nums[r], nums[i] = nums[i], nums[r]
                r = r + 1
            elif nums[i] == 2:
                nums[b], nums[i] = nums[i], nums[b]
                b = b - 1
                i = i - 1
            i = i + 1
            

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/87643008
今日推荐