75. Sort Colors

最新推荐文章于 2019-07-05 17:50:17 发布

ShenYounger

最新推荐文章于 2019-07-05 17:50:17 发布

阅读量332

点赞数

分类专栏： leetcode

leetcode 专栏收录该内容

220 篇文章

订阅专栏

本文介绍了一种解决荷兰国旗问题的方法，即在一个只包含0、1、2的数组中进行单次遍历排序的算法。该算法借鉴了快速排序的思想，通过对数组进行划分，将0置于左侧，2置于右侧，1置于中间。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

分析：题目说了一大堆，就是让对一个数组排序，数组中的元素只有0、1、2，并且要求只能遍历一遍数组，常数空间复杂度。借助于快速排序的partition思想，以1为枢纽元对数组进行划分，使0在数组的左边，2在数组的右边，1在数组的中间。通过此题，我发现自己对于快排思想理解得还不够透彻，对于快排的各种写法，尚没有熟练掌握。代码如下：

class Solution {
public:
    void sortColors(int A[], int n) {
        //zeroEnd是放0那部分的尾部索引，<span style="font-family: Arial, Helvetica, sans-serif;">twoBegin</span><span style="font-family: Arial, Helvetica, sans-serif;">是放2那部分的首部索引</span>
        //碰到0放到zeroEnd+1处，碰到2放到twoEnd-1处，碰到1指针后移
        int zeroEnd = -1, twoBegin = n, i = 0;
        while(i < twoBegin)
        {
            if(A[i] == 0 && i != ++zeroEnd)
                swap(A[zeroEnd], A[i]);
            else if(A[i] == 2 && i != --twoBegin)
                swap(A[twoBegin], A[i]);
            else
                i++;
        }
    }
};