leetcode 75. 颜色分类【双指针】

题目要求在一次遍历内完成三种颜色的分类,那么可以考虑对分类完成后位于数组两端的颜色进行分类,第三种颜色自然会落在数组中间。
使用双指针l和r,分别指向数组开头和末尾,然后从数组开头遍历数组,当数值为0时,与l指针对应的元素交换,并且同时+1,向后移位;当数值为2时,与r指针对应的元素交换。
这里要特别注意,交换完成后,r指针-1,向前移位,而由于交换后,当前元素的数值未知,因此此时遍历指针不能+1,而是继续判断该元素的大小。

**class Solution {
    public void sortColors(int[] nums) {
        int l = 0;
        int r = nums.length-1;
        int temp;
        int i=0;
        while(i<=r){
            if(nums[i] == 0){
                temp = nums[l];
                nums[l++] = nums[i];
                nums[i++] = temp;
            }else if(nums[i] == 2){
                temp = nums[r];
                nums[r--] = nums[i];
                nums[i] = temp;
            }else{
                i++;
            }
        }
    }
}**
发布了55 篇原创文章 · 获赞 0 · 访问量 772

猜你喜欢

转载自blog.csdn.net/er_ving/article/details/105055759