力扣数组中第k个最大元素

文章介绍了如何使用改进的快速排序算法在时间复杂度为O(n)的情况下找到整数数组中的第k个最大元素,避免了常规快排超出时间限制的问题。作者首先介绍了常规快速排序的思路,然后提出了一种针对寻找第k个最大值的优化版本,通过控制递归深度来提高效率。
摘要由CSDN通过智能技术生成

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1:

输入: [3,2,1,5,6,4], k = 2
输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4

提示:

  • 1 <= k <= nums.length <= 105
  • -104 <= nums[i] <= 104

思路:找到第k个最大值,那我们最先想到的是先排再找。

常见的有选择排序、冒泡排序,但对于该题会出现超出时间复杂度的问题,我们本题时间复杂度要求在O(n),所以,我们换一种排序思想,我们选择快速排序,快排时间复杂度为O(nlogn)还是超出了时间限制 ,想练一下快排算法还是写了出来,有一个测试没过。

快排思路:选择第一个作为枢纽,选用双指针i,j,从右向左先进行遍历,如果第j个位置的元素值大于第一个枢纽值,那我们j--,用while循环继续从右向左进行,如果小于,那我们就要将i与j位置元素进行交换,从左向右进行遍历,那么i++,直到i与j相遇,说明,此时i所在的位置就为我们第一个枢纽的位置,这才是进行了第一场排序,我们采用递归的思想,将枢纽左边和右边的都进行快速排序,就为快排思想。

快排代码:

int ss(int *nums,int left,int right){
    int temp;
    temp=nums[left];
    int i,j;
    i=left;
    j=right;
    while(i<j){
        while(i<j&&nums[j]<=temp){
            j--;
        }
        if(i<j){
            nums[i]=nums[j];
            i++;
        }
        while(i<j&&nums[i]>temp){
            i++;
        }
        if(i<j){
            nums[j]=nums[i];
            j--;
        }
    }
    nums[i]=temp;
    return i;
}
void Sortquick(int *nums,int left,int right){
    if(left<right){
        int i;
        i=ss(nums,left,right);
        Sortquick(nums,left,i-1);
        Sortquick(nums,i+1,right);
    }
}

本题代码:

int ss(int *nums,int left,int right){
    int temp;
    temp=nums[left];
    int i,j;
    i=left;
    j=right;
    while(i<j){
        while(i<j&&nums[j]<=temp){
            j--;
        }
        if(i<j){
            nums[i]=nums[j];
            i++;
        }
        while(i<j&&nums[i]>temp){
            i++;
        }
        if(i<j){
            nums[j]=nums[i];
            j--;
        }
    }
    nums[i]=temp;
    return i;
}
void Sortquick(int *nums,int left,int right){
    if(left<right){
        int i;
        i=ss(nums,left,right);
        Sortquick(nums,left,i-1);
        Sortquick(nums,i+1,right);
    }
}
int findKthLargest(int* nums, int numsSize, int k) {
    Sortquick(nums,0,numsSize-1);
    return nums[k-1];
}

但还是超出了时间限制:

所以本题不能用快排思想。

改进后

int quicksort(int *nums,int left,int right,int k){
    int temp=nums[left];
    int i,j;
    i=left;
    j=right;
    while(i<j){
        while(i<j&&nums[j]<=temp){
            j--;
        }
        if(i<j){
            nums[i]=nums[j];
            i++;
        }
        while(i<j&&nums[i]>temp){
            i++;
        }
        if(i<j){
            nums[j]=nums[i];
            j--;
        }
    }
    nums[i]=temp;
    if(i+1==k){
        return nums[i];
    }
    else{
        if(i+1>k){
            return quicksort(nums,left,i-1,k);
        }
        else{
            return quicksort(nums,i+1,right,k);
        }
    }
}
int findKthLargest(int* nums, int numsSize, int k) {
    int d;
    d=quicksort(nums,0,numsSize-1,k);
    return d;
}

  • 11
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
请问您需要的是两个数组的交集元素还是交集元素的数量? 如果是交集元素,可以采用双指针的方式,先将两个数组排序,然后设定两个指针 i 和 j,分别指向两个数组的开头,比较两个指针所指向的元素的大小,如果相等,则说明找到了一个交集元素,将其存到结果数组中;如果不相等,则将指向小的元素的指针向后移动一位,直到其中一个指针越界为止。 代码实现如下: ```c int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){ int i = 0, j = 0, k = 0; int* res = (int*)malloc(sizeof(int) * (nums1Size > nums2Size ? nums2Size : nums1Size)); // 排序 qsort(nums1, nums1Size, sizeof(int), cmp); qsort(nums2, nums2Size, sizeof(int), cmp); while (i < nums1Size && j < nums2Size) { if (nums1[i] == nums2[j]) { res[k++] = nums1[i]; i++; j++; } else if (nums1[i] < nums2[j]) { i++; } else { j++; } } *returnSize = k; return res; } int cmp(const void* a, const void* b) { return *(int*)a - *(int*)b; } ``` 如果是交集元素的数量,可以采用哈希表的方式,用一个哈希表来记录第一个数组中元素出现的数,然后遍历第二个数组,在哈希表中查找是否存在相同的元素,如果存在,则说明找到了一个交集元素,将其计数并从哈希表中删除该元素,最后返回计数结果即可。 代码实现如下: ```c int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){ int i = 0, j = 0, k = 0; int* res = (int*)malloc(sizeof(int) * (nums1Size > nums2Size ? nums2Size : nums1Size)); int* hash = (int*)malloc(sizeof(int) * 1001); // 由于题目条件限制在 [0, 1000] 范围内,所以哈希表可以开得比较小 memset(hash, 0, sizeof(hash)); for (i = 0; i < nums1Size; i++) { hash[nums1[i]]++; } for (j = 0; j < nums2Size; j++) { if (hash[nums2[j]] > 0) { res[k++] = nums2[j]; hash[nums2[j]]--; } } *returnSize = k; return res; } ``` 希望这些代码能够帮到您,如有疑问请随时问我~
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值