二切分问题的解法->单指针(数组访问一遍): 一个指针指向开头,依次遍历每个元素,如果是0则与指针位置的值交换并且指针后移
单指针(数组访问两遍)
使用二切分两次:先把0归位,再把1归位,最后2的位置就好了
class Solution {
public:
void sortColors(vector<int>& nums) {
int index = 0, n = nums.size();
for (int i = 0; i < n; ++i) {
if (nums[i] == 0) {
swap(nums[i], nums[index++]);
}
}
for (int i = index; i < n; ++i) {
if (nums[i] == 1) {
swap(nums[i], nums[index++]);
}
}
}
};
官方解法2->双指针(数组访问一遍)
一个指针指向开头,一个指针指向结尾,index从前往后遍历,如果是0与头指针交换头指针后移index++,2与尾指针交换尾指针前移 index不能++
易错点:
当前元素是2时,与尾指针元素交换以后index不能++,因为有可能尾指针指向的是0
class Solution {
public:
void sortColors(vector<int>& nums) {
int n = nums.size();
int front = 0, back = n - 1;
int index = 0;
while (index <= back) { //index与back相遇即可
if (nums[index] == 0) {
swap(nums[index++], nums[front++]);
} else if (nums[index] == 2){
swap(nums[index], nums[back--]);
}else {
index++;
}
}
}
};