【leetcode】75. Sort Colors

最新推荐文章于 2025-12-20 10:32:27 发布

转载最新推荐文章于 2025-12-20 10:32:27 发布 · 64 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/seyjs/p/9617370.html

文章标签：

#数据结构与算法

本文提供了一种解决荷兰国旗问题的方法，通过遍历数组并将0移至头部、2移至尾部，实现原地排序。此算法适用于颜色只有三种的情况。

题目如下：

解题思路：我的解题思路是遍历数组，遇到0删除该元素并插入到数组头部，遇到1则不处理，遇到2删除该元素并插入到数组尾部。

代码如下：

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        inx = 0
        nums += ['#']
        while inx < len(nums):
            if nums[inx] == 0:
                del nums[inx]
                nums.insert(0,0)
                inx += 1
            elif nums[inx] == 2:
                del nums[inx]
                nums.append(2)
            elif nums[inx] == '#':
                del nums[inx]
                break
            else:
                inx += 1