• 【数据结构与算法】第K大的元素:三路快速排序算法思路


    第K大的元素:三路快速排序算法思路

    Java
    https://leetcode-cn.com/problems/kth-largest-element-in-an-array/solution/di-kda-de-yuan-su-san-lu-kuai-su-pai-xu-im68n/

    解题思路

    这个题可以利用快速排序算法的隔断思想来解,我们快速排序找到的隔断(partition)在接下来的递归中都不再处理了,因此找到的隔断找到后在哪,排完序之后也是在找到的位置的,从而可以判断k与隔断的大小关系来快速得出第k大的元素在哪。首先同样的,确保第k大的数值在数组中能找到,然后第k大在数组中不好顺序表示,就直接转换成第k小的,例如第1大的可以转换成第length小的,然后因为我这边使用的三路快速排序算法(对全部一样的数值的数组有速度优势),需要传回两个值,Java不支持多变量返回,我就直接写在一个函数里了,首先是初始化好相关变量,约定好隔断:V,(l,tl)<V,[tl,i)=V,[i,tg]未处理,(tg,r)>V,然后遍历[l,r),把相对应的元素放置到对应位置,最后处理隔断V,处理好之后就变成了[l,tl)<V, [tg,r)>V, [tl,tg) = V,这样,我们再判断k在哪个区间,如果是在等于V的区间,V是已经排好序了的,那就可以直接返回,在小于V的区间就调整r值后再次找隔断,同理大于V则调整l之后再调整,这样一步一步就能得到k处的元素值。
    对三路快速排序算法不太了解的同学可以去了解一下,这边双路应该也是可以实现的。

    代码

    class Solution {
       public int findKthLargest(int[] nums, int k) {
    
            if (k > nums.length) { // 如果不存在
                return -1;
            }
    
            // 输入的是找第k大的,转换成找第k小的
             k = nums.length-k;
    
            // 为分割做准备
            Random random = new Random();
            int l = 0, r = nums.length;
    
            do {
                // 获取随机隔断
                int rd = random.nextInt(r - l) + l;
                swap(nums, rd, l);
    
                // 三路快速排序算法
                // (l,tl)<V,[tl,i)=V,[i,tg]未处理,(tg,r)>V
                int i = l + 1, tl = l + 1, tg = r - 1;
                while (i <= tg) {
                    if (nums[l] > nums[i]) {// 小于V的部分
                        swap(nums, i, tl);
                        tl++;
                        i++;
                    } else if (nums[l] < nums[i]) {
                        swap(nums, i, tg);
                        tg--;
                    } else { // 等于的部分
                        i++;
                    }
                }
                swap(nums, tl - 1, l);
                tl--;
                tg++;
    
                // [l,tl)<V,  [tg,r)>V, [tl,tg) = V
    
                if (k >= tl && k < tg) {
                    return nums[k];
                } else if (k < tl) {
                    r = tl;
                } else {
                    l = tg;
                }
            }
            while (true);
        }
    
        private void swap(int[] arr, int i, int j) {
            int t = arr[i];
            arr[i] = arr[j];
            arr[j] = t;
        }
    }
    
  • 相关阅读:
    linux安装nodejs
    Ubuntu下配置TFTP服务以及 android下使用TFTP
    笔记-《数据通信与网络教程》-第一章
    X86汇编基础-《Linux内核分析》云课堂笔记
    文章点击量排行TOP100-IBM power8算法挑战赛第三期
    LeetCode:Climbing Stairs
    LeetCode:Search for a Range
    LeetCode:Longest Substring Without Repeating Characters
    LeetCode:Linked List Cycle II
    LeetCode:Merge Sorted Array
  • 原文地址:https://www.cnblogs.com/minuy/p/15217300.html
Copyright © 2020-2023  润新知