颜色分类(双“指针”)

    科技2024-07-31  31

    题意:给定一个只包含0,1,2的数组,要求对这数组进行排序

    思路:一个简单的思路:记录数组中0,1,2的个数,之后对数组进行重新赋值。该思路需要两次遍历。下面介绍一下只进行一次遍历的方法:

    采用双“指针”,设立变量index0和index2记录0和2的位置。即区间都为0,区间都为2,其中len为数组长度。index0的初值为0,index2的初值为len-1

    先做如下设定:输入数组nums,当前遍历位置nums[i]

    解题策略如下:

    1.nums[i] == 0时,nums[i]与nums[index0]互换,之后index0++,i++

    2.nums[i] == 2 && index2 > i时, nums[i]与nums[index2]互换,之后index2--

    3.nums[i] == 1时,i++

    对于2需要做2点解释:

    1).为什么限制index2 > i?

    考虑情况:当数组已经排好序,如:[0,0,1,1,2,2]时,若不进行该限制,则会将已排好序的数组打乱。

    2).为什么不进行i++?

    因为index2所处的位置是没被遍历过的,若是进行了i++,则相当于跳过了一个元素,导致答案错误。

    代码:

    class Solution { public: void sortColors(vector<int>& nums) { int len = nums.size(); if(len <= 1) return; int index0 = 0, index2 = len - 1; for(int i=0;i<len;) { if(nums[i] == 2 && index2 > i) { nums[i] ^= nums[index2]; nums[index2] ^= nums[i]; nums[i] ^= nums[index2]; index2--; } else if(nums[i] == 0) { nums[i] ^= nums[index0]; nums[index0] ^= nums[i]; nums[i] ^= nums[index0]; index0++; i++; } else i++; } } };

     

    Processed: 0.009, SQL: 8