LeetCode:颜色分类

该博客讨论了一种不使用内置排序函数对包含红、白、蓝三种颜色的数组进行排序的方法。通过双指针策略,构建0和1的分区,确保0和1各自有序,同时保持2在最后。在遍历过程中,0和1的分区动态调整,保证了最终数组按0、1、2的顺序排序。代码实现中,使用了两个指针zero和one,分别表示0和1分区的末尾,确保正确排序。
摘要由CSDN通过智能技术生成

题目:

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

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

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

来源:力扣(LeetCode)

解:

这道题禁止使用排序函数,那么也意味着可以有别的方法。

因为颜色数字只有三个,已经确定,所以第一种方法,统计数字个数,直接重构数组,两趟O(n)完成。

这里介绍一种复杂一点的双指针算法。

我们可以想出,最后的数组肯定是3个分类,两个中间边界,那么我们就可以尝试用中间边界来构造分区。如果采用确定0和2的边界,我们就从两边向中间移动扫描,0往前面放,2往后面放,但是这里有一个bug,如果0指针遇到了1,2指针也遇到了1怎么放?如果划过又会导致分区错误,不划过就卡在这里了,所以这种方法不采用。

我们构造0和1分区,从前往后扫描,如果0就放在0分区,1就放在1分区,然后被取代的值换到扫描的i位置。2不操作(即在最后的2分区)。那么我们就需要两个分区指针,一个代表0的末尾,一个代表1分区的末尾。但是这里又有另外一个问题,0分区是动态的,势必会冲突掉1分区的头部元素,所以我们需要对每次0分区的调换操作后检查被换出的是不是1,如果是,重新调回1分区末尾,这样就确保1分区的正确性。1分区的调换不需要操作,因为扫描到第i个元素,末尾是...1(即i刚好是1的边界,相当于换自己)或...2。

class Solution {
    public void sortColors(int[] nums) {
         int zero=0,one=0;
         int t;
         for(int i=0;i<nums.length;i++){
            if(nums[i]==0){
                t=nums[zero];
                nums[zero++]=nums[i];
                nums[i]=t;
            }
            if(nums[i]==1){
                if(one<zero){
                    //初始化问题
                    one=zero;
                }
                t=nums[one];
                nums[one++]=nums[i];
                nums[i]=t;
            }
         }
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值