75. 颜色分类

最新推荐文章于 2024-09-13 16:27:15 发布

dotJunz

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量117

点赞数

分类专栏： # HOT100 文章标签：算法

本文链接：https://blog.csdn.net/m0_46283220/article/details/133974498

版权

HOT100 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

文章介绍了两种不同的方法对整数数组进行颜色分类（0,1,2），一种是利用快慢指针分别处理0和1，另一种是一次遍历并结合两个指针处理。

摘要由CSDN通过智能技术生成

75. 颜色分类

双指针——快慢指针

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;

        int p0 = 0;
        for(int i = 0; i < n; i++){
            if(nums[i] == 0){
                swap(nums, p0, i);
                p0++;
            }
        }

        int p1 = p0;
        for(int i = p0; i < n; i++){
            if(nums[i] == 1){
                swap(nums, p1, i);
                p1++;
            }
        }
    }

    void swap(int[] nums, int x, int y){
        int tmp = nums[x];
        nums[x] = nums[y];
        nums[y] = tmp;
    }
}

一次遍历

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int p0 = 0, p2 = n - 1;

        for(int i = 0; i < n; i++){
            while(i < p2 && nums[i] == 2){
                swap(nums, p2, i);
                p2--;
            }
            if(nums[i] == 0){
                swap(nums, p0, i);
                p0++;
            }
        }
    }

    void swap(int[] nums, int x, int y){
        int tmp = nums[x];
        nums[x] = nums[y];
        nums[y] = tmp;
    }
}