LeetCode-75 颜色分类常数空间三指针解法-java语言

一开始看到这道题本能反应是一道排序题，准备实现个快排，后来看到要求中有要用常数空间且只扫描一趟，想到快慢指针思路，定义了三个指针p,q,r   其中p,q指向第一个位置，r指向最后一个位置

移动规则：

（1）p当且仅当当前位置小于等于1的时候可以向右移动

（2）q当且仅当当前位置等于0的时候可以向右移动

（3）r当前仅当当前位置等于2的时候可以向左移动

算法步骤：

（1）当p位置指向的数据比r要大的时候，交换p,r数值

（2）当p此时的值为0的时候和q交换数值，目的是为了当q的位置是1的时候进行交换释放q

（3）根据移动规则移动三个指针

当p和r重叠的时候遍历排序结束

 

总结：目的是为了使得所有值为0的数据在q之前，所有值为2的数据在r之后

 

上代码

public void sortColors(int[] nums) {
        int p = 0,q = 0;
        int r = nums.length - 1;
        int temp = 0;
        while(p <= r) {
            if(nums[p] > nums[r]) {
                temp = nums[p];
                nums[p] = nums[r];
                nums[r] = temp;
            }
            
            if(nums[p] == 0){
                temp = nums[p];
                nums[p] = nums[q];
                nums[q] = temp;
            }
            
            
            if(nums[p] != 2) {
                p++;
            }
            
            if(nums[q] != 1 && nums[q] != 2) {
                q++;
            }
            if(nums[r] == 2) {
                r--;
            }
        }