LeetCode 分类练习（2）—— 三向切分 partition 思想的应用

最新推荐文章于 2022-06-29 09:59:23 发布

HeatDeath

最新推荐文章于 2022-06-29 09:59:23 发布

阅读量689

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 数据结构与算法分析（Java语言实现）文章标签： leetcode

本文为博主原创文章，未经博主允许不得转载。哈哈哈

本文链接：https://blog.youkuaiyun.com/HeatDeath/article/details/78609702

75. Sort Colors

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library's sort function for this problem.

假定数组中只有 0，1，2，对此数组进行排序。

思路：
①
1、使用一个数组统计输入数组中 0、1 、 2 的出现次数
2、对输入数组重新赋值
②
1、使用三向切分的思想，将数组中的元素按照 <1、 ==1 、>1 的顺序排列

package com.leetcode.sort;

public class SortColors {
    // 使用 计数排序
//    public void sortColors(int[] nums) {
   
   
//        int[] count = {0, 0, 0};
//        for (int num : nums)
//            count[num]++;
//        int index = 0;
//        int k = 0;
//        for (int n : count) {
   
   
//            while (n != 0) {
   
   
//                nums[index++] = k;
//                n--;
//            }
//            k++;
//        }
//    }
    // 使用 快速排序的三向切分思想
    public void sortColors(int[] nums) {
        // 假设在 nums[] 的最前端加上一个 nums[-1] = 1
        int lt = -1, i = 0, gt = nums.length;    // [0...lt] == 0  [gt...n-1] == 2
        while (i < gt) {
            if (nums[i] == 0)
                swap(nums, i++, ++lt);
            else if (nums[i] == 1)
                i++;
            else // nums[i] == 2
                swap(nums, i, --gt);
        }
    }

    public void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    public static