ali面试+荷兰国旗问题，o(n)

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <cstring>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <queue>
#define LL long long
#define inf 0x3f3f3f3f
#define mod 1e9+7
using namespace std;

void solve(vector<int>& nums)
{
    int flag1=-1,flag2=nums.size();
    for(int i=0; i<nums.size();i++){
        if(i>=flag2) break;
        if(nums[i]==0){
            flag1+=1;
            swap(nums[i], nums[flag1]);
            if(flag1!=i)i-=1;
        }
        else if(nums[i]==2){
            flag2-=1;
            swap(nums[i],nums[flag2]);
            if(flag2!=i) i-=1;
        }
    }
}

int main()
{
    vector<int> res;
    /*[2,0,2,1,1,0]*/
    res.push_back(2), res.push_back(0),res.push_back(2),res.push_back(1),res.push_back(1),res.push_back(0);
    solve(res);
    for(int i=0;i<res.size();i++){
        cout<<res[i]<<endl;
    }
    return 0;
}