算法必知--- 快速排序(拓展应用)

简介: 算法必知--- 快速排序(拓展应用)

写在前



基本思想


  • 先从数组中选择一个数作为划分值pivot,注意划分值随机选取
  • 进行区间划分的partition过程,小于pivot的放在左区间,大于pivot的放在右区间。
  • 采用递归分治的思想划分左右子区间,注意保持区间不变性


快排应用的注意点:


注意找前K大/前K小/第K大/第K小,是不需要对整个数组进行O(NlogN)的排序的!

因为可以通过快排切分直接O(N)找到第K大的数(比如求中位数就可以用本方法O(N)找到第mid大的数)


1.荷兰国旗问题



题目描述:


荷兰国旗问题:现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序。


荷兰国旗问题可以看成给定一个无序数组和目标值,将数组划分为大于、小于和等于目标值的三部分。本题实现返回等于目标值的区间。(划分值必须选中间元素)


思路分析:


  • 两个关键点:划分函数partition(使用单向扫描分区)和划分主元pivot选取(使用num)。
  • 具体实现是:定义两个边界less和more,可以把划分函数想象为向后推着大于的这部分。


代码实现:

public class NetherlandsFlag {
    public static int[] partition(int[] arr, int l, int r, int num) {
        int less = l - 1;
        int more = r + 1;
        while (l < more) {  // index指针复用l
            if (arr[l] < num) {
                swap(arr, ++less, l++);
            } else if (arr[l] > num) {
                swap(arr, --more, l);
            } else {
                l++;
            }
        }
        return new int[] {less + 1, more - 1};
    }
    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
}


75. 颜色分类,划分值选1!!

class Solution {
    public void sortColors(int[] nums) {
        int zero = -1, two = nums.length;
        int cur = 0;
        while (cur < two) {
            if (nums[cur] < 1) {
                swap(nums, ++zero, cur++);
            } else if (nums[cur] > 1) {
                swap(nums, --two, cur);
            } else {
                cur++;
            }
        }
    }
    public void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}


2.快速排序算法



思路分析:


基本实现:在荷兰国旗partition函数的基础上,我们对问题进行优化,我们可以使用递归分治的思想对左右子区间继续进行区间划分。注意:我们取每个区间的最后一个元素作为划分值pivot,为了避免时间复杂度退化(倒序情况下),我们取区间的随机值替换这个划分值。


复杂度分析:


  • 时间复杂度 O(N log N): 库函数、快排等排序算法的平均时间复杂度为 O(N log N)。
  • 空间复杂度 O(N) : 快速排序的递归深度最好(平均)为 O(log N) ,最差情况(即输入数组完全倒序)为 O(N)。


代码实现:

public static void quickSort(int[] arr) {
    if (arr == null || arr.length < 2) return;
    quickSort(arr, 0, arr.length - 1);
}
public static void quickSort(int[] arr, int l, int r) {
    if (l < r) {
        // 取随机数与最后一个数进行调换,避免极端情况复杂度退化
        swap(arr, l + (int)(Math.random() * (r - l + 1)), r);
        int[] p = partition(arr, l, r);
        quickSort(arr, l, p[0] - 1);
        quickSort(arr, p[1] + 1, r);
    }
}
public static int[] partition(int[] arr, int l, int r) {
    int less = l - 1;
    int more = r;
    while (l < more) {
        if (arr[l] < arr[r]) {
            swap(arr, ++less, l++);
        } else if (arr[l] > arr[r]) {
            swap(arr, --more, l);
        } else {
            l++;
        }
    }
    // 把划分主元放到正确的位置
    swap(arr, more, r);
    return new int[] {less + 1, more};
}
public static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}


3.快排进阶(应用)



描述:求解未排序数组中第K大的元素(leetcode215)/前K大个元素。


思路分析:求解第k大的元素,借助 partition 操作定位到最终排定以后索引为 nums.length - k 的那个元素(特别注意:随机化切分元素),根据划分值继续决定结果所在区间,注意每次划分要随机选取划分值pivot(本题取区间随机值与区间尾元素交换,作为划分值),否则会严重影响效率。


代码实现:数组中第k大的元素

public int findKthLargest(int[] nums, int k) {
    if (nums == null || nums.length == 0 || k == 0) return 0;
    int target = nums.length - k;
    return quickSort(nums, 0, nums.length - 1, target);
}
private int quickSort(int[] nums, int l, int r, int target) {
    int index = partition(nums, l, r);
    if (index == target) return nums[index];
    return index > target ? quickSort(nums, l, index - 1, target) : quickSort(nums, index + 1, r, target);
}
private int partition(int[] nums, int l, int r) {
    // 一定要随机选取划分值!
    swap(nums, l + (int)(Math.random() * (r - l + 1)), r);
    int less = l - 1, more = r;
    while (l < more) {
        if (nums[l] < nums[r]) {
            swap(nums, ++less, l++);
        } else if (nums[l] > nums[r]) {
            swap(nums, --more, l);
        } else {
            l++;
        }
    }
    swap(nums, more, r);
    return more;
}
public void swap(int[] nums, int i, int j) {
    int temp = nums[i];
    nums[i] = nums[j];
    nums[j] = temp;
}


partion划分函数也可以进行代码简化,原理相同。

private int partition(int[] nums, int l, int r) {
    swap(nums, l + (int)(Math.random() * (r - l + 1)), r);
    int less = l - 1, pivot = nums[r];
    for (int i = l; i < r; ++i) {
        if (nums[i] <= pivot) {
            swap(nums, ++less, i);
        }
    }
    swap(nums, less + 1, r);
    return less + 1;
}


描述:找到未排序数组的前k小个元素(剑指offer40)/第K小个元素。


思路:本题与上题思路相同,只不过注意两点:


  • 注意:边界条件判断,basecase中k == 0情况!
  • 第k小对应的索引值为k - 1。


代码

public int[] getLeastNumbers(int[] arr, int k) {
    if (k == 0 || arr == null || arr.length == 0) return new int[0];
    // 注意:第k小的元素下标为k - 1
    return quickSort(arr, 0, arr.length - 1, k - 1);
}
private int[] quickSort(int[] arr, int l, int r, int k) {
    int index = partition(arr, l, r);
    if (index == k) {
        // 复制数组arr的前k个数,也就是0 - k-1
        return Arrays.copyOf(arr, index + 1);
    }
    return index > k ? quickSort(arr, l, index - 1, k) : quickSort(arr, index + 1, r, k);
}
private int partition(int[] arr, int l, int r) {
    // 随机选取划分值
    swap(arr, l + (int)(Math.random() * (r - l + 1)), r);
    int less = l - 1, pivot = arr[r];
    for (int i = l; i < r; ++i) {
        if (arr[i] <= pivot) {
            swap(arr, ++less, i);
        }
    }
    swap(arr, less + 1, r);
    return less + 1;
}
public void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}


相关文章
|
16天前
|
搜索推荐 C语言
【排序算法】快速排序升级版--三路快排详解 + 实现(c语言)
本文介绍了快速排序的升级版——三路快排。传统快速排序在处理大量相同元素时效率较低,而三路快排通过将数组分为三部分(小于、等于、大于基准值)来优化这一问题。文章详细讲解了三路快排的实现步骤,并提供了完整的代码示例。
42 4
|
1月前
|
存储 算法 Java
解析HashSet的工作原理,揭示Set如何利用哈希算法和equals()方法确保元素唯一性,并通过示例代码展示了其“无重复”特性的具体应用
在Java中,Set接口以其独特的“无重复”特性脱颖而出。本文通过解析HashSet的工作原理,揭示Set如何利用哈希算法和equals()方法确保元素唯一性,并通过示例代码展示了其“无重复”特性的具体应用。
44 3
|
1月前
|
机器学习/深度学习 人工智能 自然语言处理
深度学习中的优化算法及其应用
【10月更文挑战第8天】 本文将探讨深度学习中常用的优化算法,包括梯度下降法、Adam和RMSProp等,介绍这些算法的基本原理与应用场景。通过实例分析,帮助读者更好地理解和应用这些优化算法,提高深度学习模型的训练效率与性能。
149 63
|
20天前
|
机器学习/深度学习 JSON 算法
二叉树遍历算法的应用场景有哪些?
【10月更文挑战第29天】二叉树遍历算法作为一种基础而重要的算法,在许多领域都有着不可或缺的应用,它为解决各种复杂的问题提供了有效的手段和思路。随着计算机科学的不断发展,二叉树遍历算法也在不断地被优化和扩展,以适应新的应用场景和需求。
24 0
|
1月前
|
存储 算法 搜索推荐
这些算法在实际应用中有哪些具体案例呢
【10月更文挑战第19天】这些算法在实际应用中有哪些具体案例呢
38 1
|
1月前
|
机器学习/深度学习 人工智能 算法
[大语言模型-算法优化] 微调技术-LoRA算法原理及优化应用详解
[大语言模型-算法优化] 微调技术-LoRA算法原理及优化应用详解
81 0
[大语言模型-算法优化] 微调技术-LoRA算法原理及优化应用详解
|
1月前
|
算法 搜索推荐 Shell
数据结构与算法学习十二:希尔排序、快速排序(递归、好理解)、归并排序(递归、难理解)
这篇文章介绍了希尔排序、快速排序和归并排序三种排序算法的基本概念、实现思路、代码实现及其测试结果。
23 1
|
1月前
|
算法 安全 物联网
如何应用SM2算法进行身份认证
【10月更文挑战第5天】如何应用SM2算法进行身份认证
73 1
|
1月前
|
监控 算法 数据挖掘
HyperLogLog算法有哪些应用场景呢
【10月更文挑战第19天】HyperLogLog算法有哪些应用场景呢
17 0
|
1月前
|
机器学习/深度学习 算法 数据建模
计算机前沿技术-人工智能算法-生成对抗网络-算法原理及应用实践
计算机前沿技术-人工智能算法-生成对抗网络-算法原理及应用实践
30 0
下一篇
无影云桌面