《LeetCode之每日一题》:86.数据流中的第 K 大元素

Posted 是七喜呀!

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了《LeetCode之每日一题》:86.数据流中的第 K 大元素相关的知识,希望对你有一定的参考价值。

数据流中的第 K 大元素


题目链接: 数据流中的第 K 大元素

有关题目

设计一个找到数据流中第 k 大元素的类(class)。
注意是排序后的第 k 大元素,不是第 k 个不同的元素。
请实现 KthLargest 类:

KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。

int add(int val) 将 val 插入数据流 nums 后,返回当前数据流中第 k 大的元素。
示例:

输入:
["KthLargest", "add", "add", "add", "add", "add"]
[[3, [4, 5, 8, 2]], [3], [5], [10], [9], [4]]
输出:
[null, 4, 5, 5, 8, 8]

解释:
KthLargest kthLargest = new KthLargest(3, [4, 5, 8, 2]);
kthLargest.add(3);   // return 4
kthLargest.add(5);   // return 5
kthLargest.add(10);  // return 5
kthLargest.add(9);   // return 8
kthLargest.add(4);   // return 8
提示:
1 <= k <= 10^4
0 <= nums.length <= 10^4
-104 <= nums[i] <= 10^4
-10^4 <= val <= 10^4
最多调用 add 方法 10^4 次
题目数据保证,在查找第 k 大元素时,数组中至少有 k 个元素

题解

法一:优先队列
参考动画理解Hao Kun Yang
参考负雪明烛
参考TopK理解
参考快排算法

struct Heap {
    int* heap;
    int heapSize;
    bool (*cmp)(int, int);
};//优先队列

//初始化,优先队列
void init(struct Heap* obj, int n, bool (*cmp)(int, int)) {
    obj->heap = malloc(sizeof(int) * (n + 1));//元素下标从1开始,默认的[0]我们不使用
    obj->heapSize = 0;//初始化值为0
    obj->cmp = cmp;//函数指针
}

//比较函数
bool cmp(int a, int b){
    return a > b;
}

//使用指针交换
void swap(int* a, int* b){
    int tmp = *a;
    *a = *b, *b = tmp;
}

//插入元素
void push(struct Heap* obj, int x){
    int p = ++(obj->heapSize), q = p >> 1;//根据小根堆,节点之间的关系
    obj->heap[p] = x;//插入最后一个元素
    while (q){
        if (!obj->cmp(obj->heap[q], obj->heap[p])) {
            break;
        }
        swap(&(obj->heap[q]), &(obj->heap[p]));//子父节点交换
        p = q, q = p >> 1;//子父节点对应下标值,改变
    }    
}

//弹出堆顶
void pop(struct Heap* obj) {
    swap(&(obj->heap[1]), &(obj->heap[(obj->heapSize)--]));
    int p = 1, q = p << 1;
    while (q <= obj->heapSize) {
        if (q + 1 <= obj->heapSize) {
            if (obj->cmp(obj->heap[q], obj->heap[q + 1])) {
                q++;//这边有点模糊
            }
        }
        if (!obj->cmp(obj->heap[p], obj->heap[q])) {
            break;
        }
        swap(&(obj->heap[q]), &(obj->heap[p]));
        p = q, q = p << 1;
    }
}

//弹出队头元素
int top(struct Heap* obj) {
    return obj->heap[1];
}

typedef struct {
    struct Heap* heap;
    int maxSize;
} KthLargest;


KthLargest* kthLargestCreate(int k, int* nums, int numsSize) {
    KthLargest* ret = malloc(sizeof(KthLargest));
    ret->heap = malloc(sizeof(struct Heap));
    init(ret->heap, k + 1, cmp);
    ret->maxSize = k;
    for (int i = 0; i < numsSize; i++) {
        kthLargestAdd(ret, nums[i]);
    }
    return ret;
}

int kthLargestAdd(KthLargest* obj, int val) {
    push(obj->heap, val);
    if (obj->heap->heapSize > obj->maxSize) {
        pop(obj->heap);
    }
    return top(obj->heap);
}

void kthLargestFree(KthLargest* obj) {
    free(obj->heap->heap);//释放动态开辟数组
    free(obj->heap);//释放动态开辟的队列
    free(obj);//释放动态开辟数据流
}

/**
 * Your KthLargest struct will be instantiated and called as such:
 * KthLargest* obj = kthLargestCreate(k, nums, numsSize);
 * int param_1 = kthLargestAdd(obj, val);
 
 * kthLargestFree(obj);
*/

在这里插入图片描述

以上是关于《LeetCode之每日一题》:86.数据流中的第 K 大元素的主要内容,如果未能解决你的问题,请参考以下文章

《LeetCode之每日一题》:130.字符串中的第一个唯一字符

《LeetCode之每日一题》:97.数组中的第K个最大元素

《LeetCode之每日一题》:90.在排序数组中查找元素的第一个和最后一个位置

《LeetCode之每日一题》:242.一年中的第几天

《LeetCode之每日一题》:133.删除链表中的节点

《LeetCode之每日一题》:36.外观数列