(lintcode)第5题第k大元素

在数组中找到第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;inums[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);//递归调用排序基数后面一半
        }
    }
};




你可能感兴趣的:(lintcode)