题意:给定一个只包含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++; } } };