【算法系列】快排优化/堆排序重复数据

题目:912. 排序数组 - 力扣(LeetCode)

题解

  1. 快速排序可以通过随机选取基准值来优化一些较极端的情况,使得时间复杂度控制在O(NlogN)
  2. 堆排序是可以处理的重复数据的哦

代码

class Solution {
public:
    vector<int> sortArray(vector<int>& nums) {
        // 堆排序 创建大根堆 向下调整法
        HeapSort(nums);
        return nums;
    }

    void AdjustDown(vector<int>& nums, int pos, int size)
    {
        int parent = pos;
        int child = 2 * parent + 1;
        if(child + 1 < size) child = nums[child] > nums[child + 1] ? child : child + 1;

        while(child < size)
        {
            if(nums[parent] < nums[child])
            {
                std::swap(nums[parent], nums[child]);
                parent = child;
                child = 2 * parent + 1;
                if(child + 1 < size) child = nums[child] > nums[child + 1] ? child : child + 1;
            }
            else break;
        }
    }

    void HeapSort(vector<int>& nums)
    {
        int size = nums.size();
        // 创建大根堆
        int lastParent = (nums.size() - 2) / 2;
        for(int i = lastParent; i >= 0; i--)
        {
            AdjustDown(nums, i, size);
        }

        // 堆排序
        for(int i = 0; i < size - 1; i++)
        {
            std::swap(nums[0], nums[size - 1 - i]);
            AdjustDown(nums, 0, size - 1 - i);
        }
    }
};
class Solution {
public:
    vector<int> sortArray(vector<int>& nums) {
        // 快速排序:随机选取基准数进行优化
        srand(time(nullptr));
        QuickSort(nums, 0, nums.size() - 1);
        return nums;
    }

    void RandomOptimize(vector<int>& nums, int left, int right)
    {
        int index = rand() % (right - left + 1) + left;
        std::swap(nums[left], nums[index]);
    }

    void QuickSort(vector<int>& nums, int left, int right)
    {
        if(left >= right) return;
        RandomOptimize(nums, left, right);
        int key = nums[left];
        int l = left, r = right;
        while(l < r)
        {
            while(r > l && nums[r] >= key) r--;
            if(r == l) break;
            while(l < r && nums[l] <= key) l++;
            if(l < r) std::swap(nums[l], nums[r]);
        }
        std::swap(nums[l], nums[left]);

        QuickSort(nums, left, l - 1);
        QuickSort(nums, l + 1, right);
    }
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值