实战1: 三指针分治很简单

颜色分类

#include <vector>
#include <iostream>

void swap(int& a, int& b)
{
  int temp;
  temp = a;
  a = b;
  b = temp;
}
    

void sortColors(std::vector<int>& nums) 
{
        // left：0区间右边界，i：当前遍历元素，right：2区间左边界
        int left = -1, i = 0, right = nums.size();
        // 遍历终止条件：i进入2区间（所有元素处理完毕）
        while(i < right)
        {
            if(nums[i] == 0)
            {
                // 0归入0区间，left右移，i右移（当前元素已处理）
                swap(nums[++left], nums[i++]);
            } 
            else if(nums[i] == 1) 
            {
                // 1无需交换，直接i右移（归入1区间）
                i++;
            } 
            else 
            {
                // 2归入2区间，right左移，i不右移（新元素待判断）
                swap(nums[--right], nums[i]);
            }
        }
}

int main()
{
    std::vector<int> nums { 1, 0, 1, 0, 2, 1, 0, 1, 0, 2};

    sortColors(nums);

    for(int i : nums)
        std::cout << i << '\t';
    
}

这个是我看了Q741_147大佬理解的,而运行也是正确的:

• i(当前元素)只走一遍就可以实现;
• left和right向中间缩,为了让它们走,所以++left,- -right
• left所走过的就是0区, right所走过的就是2区,
• i遇到0的时候和left换,
• 遇到1的时候继续走,
• 遇到2的时候和right换, i不变.

left在容器左端 left = -1;
right 在容器右端 left = nums.size();
i从容器里面的首元素开始 i = 0;

int left = -1, i = 0, right = nums.size();

为了保证left和right向中间缩, 所以下面都是++left,- -right
i开始走了,有三种情况:
遇到0的时候和left换:

if (nums[i] == 0) swap(nums[++left], nums[i++]);

遇到1的时候继续走

if (nums[i] == 1) ++i;

遇到2的时候和right换, i待着不动

if (nums[i] == 2) swap(nums[++right], nums[i]);

记忆口诀:
父母(left)是我(i)的起点(左端-1), 一把屎一尿(0)养我. 先让父母走(nums[++left]),自己稍后走(nums[i++]),这是礼节;

和妻子(right)白头偕老是我的终点(nums.size()), 初恋时记得给她拿好东西(2), 看着她走了(nums[++right])自己还呆呆地望着(nums[i])

中间[left, right]不好也不坏(1)的时候是自己走(++i).