LeetCode刷题整理——二叉树层序遍历，TOP K

最新推荐文章于 2025-01-21 20:19:54 发布

原创最新推荐文章于 2025-01-21 20:19:54 发布 · 347 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #leetcode #算法

算法专栏收录该内容

14 篇文章

订阅专栏

本文介绍了二叉树层序遍历的实现技巧，通过计算队列中元素数量来标记层序遍历的结束。此外，还探讨了TOPK问题的解决方法，包括使用堆、快速选择等算法，并提供了三种不同的快速选择实现方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

二叉树层序遍历

要标记一层遍历的结束，可以先计算出当前队列里面有多少个元素，然后一次性处理完这些元素就是处理完了一层。
原来的方法是用一个指针指向下一层的结尾（不好写）。

TOP K

堆（优先队列）、快速选择。

LeetCode 215
LeetCode 剑指offer 40

用传统的quick select写法做不出来，还有，pivot选最左端的也做不出来（？到底为什么。。可能我菜写法不对）。选右端，最好随机选（因为极端用例会超时）。改进的话，可以用双指针搜索、交换，一个工作指针负责遍历，另一个指向符合条件（大于或小于基准值）区的末尾。
方法1：

	// 快排切分，返回下标j，使得比nums[j]小的数都在j的左边，比nums[j]大的数都在j的右边。
    private int partition(int[] nums, int lo, int hi) {
        int v = nums[lo];
        int i = lo, j = hi + 1;
        while (true) {
            while (++i <= hi && nums[i] < v);
            while (--j >= lo && nums[j] > v);
            if (i >= j) {
                break;
            }
            int t = nums[j];
            nums[j] = nums[i];
            nums[i] = t;
        }
        nums[lo] = nums[j];
        nums[j] = v;
        return j;
    }

方法2：

	int partition(vector<int> &nums,int left,int right){
        int pivot=nums[right];
        int j=left,i=left;
        for(;i<right;i++){
            if(nums[i]>=pivot){
                swap(nums[i],nums[j]);
                j++;
            }
        }
        swap(nums[j],nums[right]);
        return j;
    }

方法3：

	int partition(vector<int> &nums,int left,int right){
        // int pivot=nums[right];
        srand((unsigned) time(NULL));
        //随机选[left, right]之间的随机整数
	    int pivot_pos = (rand() % (right - left + 1)) + left;
	    int pivot = nums[pivot_pos];
        swap(nums[right],nums[pivot_pos]);
        int j=left,i=left;
        for(;i<right;i++){
            if(nums[i]>=pivot){
                swap(nums[i],nums[j]);
                j++;
            }
        }
        swap(nums[j],nums[right]);
        return j;
    }