时间复杂度和空间复杂度击败了80%的人!
看到题目,有些同学就想问:哎!这道题不是很简单嘛,排序一下再数组寻找坐标不就完事儿了嘛
这么想就不对了哦,因为我们只需要求第k大的数字,而排序是把每个数字都排序,所以直接对数组进行排序的话时间复杂度就会很高
思路:这里用的是堆排序,首先把数组变成大顶堆数组(把数组看成一个堆,依次根据数组下标排成二叉树,再对二叉树进行大顶堆化,这样这个数组就是一个大顶堆啦),变成之后,数组第一个位置就是最大的位置,我们把这个位置和数组最后一个位置进行交换,再对第一个位置进行大顶堆化。
代码如下
class Solution {
public int findKthLargest(int[] nums, int k) {
int len=nums.length;
//先构造大顶堆
for(int i=len/2;i>0;i--){
heapSort(nums,i,len);
}
//找出第k大的数字
for(int i=0;i<k;i++){
if(i+1==k) return nums[0];
int temp=nums[0];
nums[0]=nums[len-1-i];
nums[len-1-i]=temp;
heapSort(nums,1,len-1-i);
}
return -1;
}
//这个函数的意思就是对二叉树中的位置i
//依次和它的左孩子和右孩子比较,孩子大的话就和孩子交换顺序
//交换完之后,还需要对这个位置递归,因为要保证每一个节点都是大顶堆
//左孩子:2*i;右孩子:2*i+1(数组会-1)
public void heapSort(int[] nums,int i,int n){
int j=2*i;
if(j<=n){
if(j+1<=n){
if(nums[j-1]<nums[j]) j+=1;
}
if(nums[j-1]>nums[i-1]){
int temp=nums[j-1];
nums[j-1]=nums[i-1];
nums[i-1]=temp;
heapSort(nums,j,n);
}
}
}
}