【LeetCode热题100】--75.颜色分类

最新推荐文章于 2024-10-05 12:29:44 发布

小d<>

最新推荐文章于 2024-10-05 12:29:44 发布

阅读量277

点赞数

分类专栏： LeetCode # hot100 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_46656857/article/details/133895045

版权

LeetCode 同时被 2 个专栏收录

91 篇文章 2 订阅

订阅专栏

hot100

58 篇文章 0 订阅

订阅专栏

75.颜色分类

方法一：使用单指针

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int ptr = 0;
        for(int i = 0;i<n;i++){
            if(nums[i] == 0)
            {
                int tmp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = tmp;
                ++ptr;
            }
        }
        for(int i = ptr;i<n;i++){
            if(nums[i] == 1){
                int tmp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = tmp;
                ++ptr;
            }
        }

    }
}

方法二：使用双指针

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int p0 = 0, p2 = n - 1;
        for (int i = 0; i <= p2; ++i) {
            while (i <= p2 && nums[i] == 2) {
                int temp = nums[i];
                nums[i] = nums[p2];
                nums[p2] = temp;
                --p2;
            }
            if (nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[p0];
                nums[p0] = temp;
                ++p0;
            }
        }
    }
}