算法思想:
该问题是一个排序问题,可以将这个数组分为前部、中部、后部。这里的思路如下:将前部和后部的排在数组的前边和后边,中部的就自然排好了。设置三个指针:begin,mid,end,然后从mid的位置开始遍历。
1.如果遍历到的位置为0,则说明它属于前部,就将mid和begin交换,然后mid和begin前进。
2.如果遍历到的位置是1,则说明它属于中部,则只将mid前进。
3.如果遍历到的位置是2,则说明它属于后部,将mid和end交换,由于交换后mid的指向不确定,因此只将end--。
void sortColors(vector<int>& nums) { int begin=0,end=nums.size()-1,mid=0; while(mid<=end&&begin<=mid){ if(nums[mid]==0){ swap(nums[mid],nums[begin]); begin++; mid++; } else if(nums[mid]==1) mid++; else if(nums[mid]==2){ swap(nums[mid],nums[end]); end--; } } }
时间: 2024-11-09 05:47:17