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