使用双路快排实现颜色排序疑问

老师你好，我仿照着双路排序尝试解决颜色排序的问题。

对于第一版代码

    public void sortColors(int[] nums){
        int targetValue = 1;
        int zeroBorder = 0;
        int twoBorder = nums.length - 1;

        while (true){
            while (zeroBorder <= twoBorder && nums[zeroBorder] < targetValue)
                zeroBorder++;

            while (twoBorder >= zeroBorder && nums[twoBorder] > targetValue)
                twoBorder--;

            if (zeroBorder >= twoBorder) break;

            swap(nums,zeroBorder,twoBorder);

            zeroBorder++;
            twoBorder--;
        }
    }

发现测试用例 [2,0,1] 无法通过，跟踪代码后，发现是最后的twoBorder对应的元素没有进行交换。我想是不是这种排序方式需要在所有循环结束后，进行一次赋值，于是我把代码改成了这样

public static void sortColors(int[] nums){
        int targetValue = 1;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == targetValue){
                swap(nums,0,i);
                break;
            }
        }

        int zeroBorder = 1;
        int twoBorder = nums.length - 1;

        while (true){
            while (zeroBorder <= twoBorder && nums[zeroBorder] < targetValue)
                zeroBorder++;

            while (twoBorder >= zeroBorder && nums[twoBorder] > targetValue)
                twoBorder--;

            if (zeroBorder >= twoBorder) break;

            swap(nums,zeroBorder,twoBorder);

            zeroBorder++;
            twoBorder--;
        }

        swap(nums, 0, twoBorder);
    }

但是这种方式，测试用例[2,0,2,1,1,0] 无法通过。在跟踪代码的过程中逐渐迷惑了，想问问老师这种方式可以实现颜色排序吗