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

本文介绍了二叉树层序遍历的实现技巧,通过计算队列中元素数量来标记层序遍历的结束。此外,还探讨了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;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值