三色排序

最新推荐文章于 2024-01-29 14:50:18 发布

原创最新推荐文章于 2024-01-29 14:50:18 发布 · 424 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

数据结构数组和矩阵同时被 2 个专栏收录

20 篇文章

订阅专栏

数据结构排序

10 篇文章

订阅专栏

本文介绍了一种基于快速排序思想的三色排序算法，该算法能够有效地对只包含0、1、2三种元素的整数数组进行原地排序，无需额外计数。通过对数组中的元素进行扫描和交换操作实现排序。

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

有一个只由0，1，2三种元素构成的整数数组，请使用交换、原地排序而不是使用计数进行排序。

给定一个只含0，1，2的整数数组A及它的大小，请返回排序后的数组。保证数组大小小于等于500。

测试样例：

[0,1,1,0,2,2],6

返回：[0,0,1,1,2,2]

思路：与快排划分过程相似

class ThreeColor {
public:
    vector<int> sortThreeColor(vector<int> A, int n) {
         // 利用快排思想
        int low=0,high = n-1;
        for(int i = 0;i<=high;i++){
            if(A[i]==0) {
                swap(A[i],A[low]);
                low ++;
            }else if(A[i]==2){
                swap(A[i],A[high]);
                high--;
                i--;//交换过来的A[high]未知，需要i从新比较一次
            }
        }
        return A;
    }
};