刷题笔记 Hot100 75. 颜色分类

春种一粒粟秋收一粒米

于 2022-10-10 16:27:38 发布

阅读量133

点赞数

CC 4.0 BY-SA版权

文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/WuLiang_995/article/details/127247178

本文介绍了一种使用三个指针i、j、k实现的颜色排序算法，该算法能够原地将数组中的0、1、2三种元素按顺序排列。通过跟踪每个数字的最后一个已排序位置，并采用数组插入而非直接交换的方法来移动元素。

利用三个下标，最后一个0的待插入下标为i，最后一个1的待插入下标为j，最后一个2的待插入下标为k
思想也很简单，对于原地移动的数组，需要考虑下标和数据之间的关系
如果是0，就要移动到第 i 个位置，同时 i + 1
如果是1，需要移动到第 i + j 个位置，同时 j + 1
如果是2，需要移动到第 i + j + k 个位置，同时 k + 1
这里的移动采用数组插入的方式，不能直接交换！

class Solution {
    public void sortColors(int[] nums) {
        int i = 0, j = 0, k = 0, length = nums.length;
        for (int index = 0; index < length; index++) {
            switch (nums[index]) {
                case 0:
                    exchange(nums, i++, index);
                    break;
                case 1:
                    exchange(nums, i + j++, index);
                    break;
                case 2:
                    exchange(nums, i + j + k++, index);
                    break;
            }
        }
    }

    private void exchange(int[] nums, int i, int j) {
        int num = nums[j];
        for (int k = j; k > i; k--) {
            nums[k] = nums[k - 1];
        }
        nums[i] = num;
    }
}