(lintcode)第5题第k大元素

来源:互联网 发布:淘宝历史版本 编辑:程序博客网 时间:2024/04/30 03:28

在数组中找到第k大的元素(可以交换数组中的元素的位置)

样例:

给出数组 [9,3,2,4,8],第三大的元素是 4

给出数组 [1,2,3,4,5],第一大的元素是 5,第二大的元素是 4,第三大的元素是 3,以此类推

首先想到的是最简单的冒泡排序,但是这样的话,特别耗时间,最后的测试数据很多的时候,跑得特别慢,超时了。在这里还是放一下代码:

class Solution {    /*     * @param k : description of k     * @param nums : array of nums     * @return: description of return     */    public int kthLargestElement(int k, int[] nums) {        // write your code here        for(int i=0;i<k;i++){            for(int j=0;j<nums.length-1-i;j++){                if(nums[j]>nums[j+1])                {                    int temp=nums[j];                    nums[j]=nums[j+1];                    nums[j+1]=temp;                }            }        }        return nums[nums.length-k];    }};

另一种解法就是使用快速排序,先全部按照顺序排好,再找出第k大的数,这样的耗时比较少。(至于快速排序会在另一篇blog中分析一下)

class Solution {    /*     * @param k : description of k     * @param nums : array of nums     * @return: description of return     */public static int kthLargestElement(int k, int[] nums) {        quickstart(0,nums.length-1,nums);        return nums[nums.length-k];    }    public static void quickstart(int l,int r,int[]nums){        if (l < r)        {            //Swap(s[l], s[(l + r) / 2]); //将中间的这个数和第一个数交换             int i = l, j = r, temp = nums[l];            while (i < j)            {                while(i < j && nums[j] >= temp) // 从右向左找第一个小于temp的数                    j--;                if(i < j)                    nums[i++] = nums[j];                while(i < j && nums[i] < temp) // 从左向右找第一个大于等于temp的数                    i++;                if(i < j)                    nums[j--] = nums[i];            }            nums[i] = temp;            quickstart( l, i - 1,nums); // 递归调用排序基数前面一半            quickstart( i + 1, r,nums);//递归调用排序基数后面一半        }    }};




原创粉丝点击