LeetCode第75题颜色分类

Ealiser

已于 2022-03-21 13:49:49 修改

阅读量646

点赞数

分类专栏：数据结构数组文章标签： leetcode 算法职场和发展

于 2022-03-21 13:48:35 首次发布

本文链接：https://blog.csdn.net/ealiser/article/details/123633773

版权

数据结构同时被 2 个专栏收录

40 篇文章 0 订阅

订阅专栏

数组

30 篇文章 0 订阅

订阅专栏

该博客讨论了在有限空间条件下，如何选择排序算法。作者选择了堆排序作为解决方案，并提供了具体的Java代码实现。堆排序算法通过维护一个最大（或最小）堆来逐步排序数组，其时间复杂度为O(n log n)。代码中展示了如何构建和调整堆以达到排序目的。

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

思想
堆排序
核心算法
要求为使用常数空间，则符合要求的有冒泡排序、选择排序、插入排序、希尔排序、堆排序。本次采用堆排序。因为限制为2，计数排序应该效果也不错。
代码

class Solution {
    public static void swap(int[] a, int i, int j){
        int temp;
        temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }
    public void sortColors(int[] nums) {
        int len = nums.length;
        int root = 0;
        int i;
        int j = len - 1;
        while(j > 0){
            i = j;
            while (i > 0){
                if((i-2)%2 == 1 || i ==1) {
                    root = (i - 1) / 2;
                    i -= 1;
                    if(nums[root] < nums[root*2 + 1]) swap(nums,root,root*2+1);
                }
                else {
                    root = (i - 2) / 2;
                    i -= 2;
                    if(nums[root*2+1] < nums[root*2+2]){
                        if(nums[root] < nums[root*2+2]) swap(nums,root,root
                                *2+2);
                    }else if(nums[root] < nums[root*2+1]) swap(nums,root,root
                            *2+1);
                }
            }
            swap(nums,0,j);
            j--;
                }

        
    }
}