LeetCode 75. 颜色分类(JAVA)

题目

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

解题思路

public void sortColors(int[] nums) {
        if (nums.length <= 1){
            return;
        };
//        //双指针
//        for (int left =0;left<nums.length;left++){
//            for (int right = left+1;right< nums.length;right++){
//                if (nums[left]>nums[right])
//                {
//                    int temp = nums[left];
//                    nums[left] = nums[right];
//                    nums[right] = temp;
//                }
//            }
//        }
        //三指针
        if (nums.length == 2 && nums[0] > nums[1])
        {
            this.swap(0,1,nums);
            return;
        }
        int left = 0;
        int right = nums.length;
        int cur = 0;
        while (cur < right){
            System.out.println(Arrays.toString(nums));
            if (nums[cur] == 0){
                swap(left, cur , nums);
                left++;
                cur++;
            }
            else if (nums[cur] == 1){
                cur++;
            }
            else{
                right--;
                swap(cur, right, nums);
            }
        }
        return;
    }
    public void swap (int nums1, int nums2, int[] nums) {
        int temp = nums[nums1];
        nums[nums1] = nums[nums2];
        nums[nums2] = temp;
    }