leetcode刷题记录37-2476. 二叉搜索树最近节点查询

问题描述

给你一个 二叉搜索树 的根节点 root ,和一个由正整数组成、长度为 n 的数组 queries

请你找出一个长度为 n二维 答案数组 answer ,其中 answer[i] = [mini, maxi]

  • mini 是树中小于等于 queries[i]最大值 。如果不存在这样的值,则使用 -1 代替。
  • maxi 是树中大于等于 queries[i]最小值 。如果不存在这样的值,则使用 -1 代替。

返回数组 answer

示例

示例 1 :

复制代码
输入:root = [6,2,13,1,4,9,15,null,null,null,null,null,null,14], queries = [2,5,16]
输出:[[2,2],[4,6],[15,-1]]
解释:按下面的描述找出并返回查询的答案:
- 树中小于等于 2 的最大值是 2 ,且大于等于 2 的最小值也是 2 。所以第一个查询的答案是 [2,2] 。
- 树中小于等于 5 的最大值是 4 ,且大于等于 5 的最小值是 6 。所以第二个查询的答案是 [4,6] 。
- 树中小于等于 16 的最大值是 15 ,且大于等于 16 的最小值不存在。所以第三个查询的答案是 [15,-1] 。

示例 2 :

复制代码
输入:root = [4,null,9], queries = [3]
输出:[[-1,4]]
解释:树中不存在小于等于 3 的最大值,且大于等于 3 的最小值是 4 。所以查询的答案是 [-1,4] 。

提示:

  • 树中节点的数目在范围 [2, 10^5]
  • 1 <= Node.val <= 10^6
  • n == queries.length
  • 1 <= n <= 10^5
  • 1 <= queries[i] <= 10^6

问题分析:

刚做完一看运行时间1966ms,击败了19.57%。心说完了,自己想出来的无脑解法果然废了。。。然后一看官解,好好好,官解剽窃我的思路是吧。

我的思路就是:首先题目给了一个二叉搜索树,我们直接中序遍历把树节点值全存一个数组arr里,易知这个数组有序,接着直接二分查找对应结点即可。

代码如下:

cpp 复制代码
class Solution {
public:
    vector<vector<int>> closestNodes(TreeNode* root, vector<int>& queries) {
        vector<int> res;
        //    中序遍历存储有序数组
        function<void(TreeNode*)> inorder = [&](TreeNode* node) {
            if (node == nullptr)
                return;
            inorder(node->left);
            res.push_back(node->val);
            inorder(node->right);
        };
        inorder(root);
        //    结果数组
        vector<vector<int>> ans;
        for (int i = 0; i < queries.size(); ++i) {
            //    内置二分找大于等于元素,返回迭代器
            auto p = lower_bound(res.begin(), res.end(), queries[i]);
            //    如果找到了并且是对应元素本身
            if (p != res.end() && *p == queries[i]) {
                ans.push_back({*p, *p});
            //    否则push{lower,upper}
            } else {
                int lower = (p == res.begin()) ? -1 : *(p - 1);
                int upper = (p == res.end()) ? -1 : *p;
                ans.push_back({lower, upper});
            }
        }
        return ans;
    }
};
相关推荐
光而不耀@lgy1 分钟前
C++初登门槛
linux·开发语言·网络·c++·后端
啊丢_12 分钟前
C++——Lambda表达式
开发语言·c++
CODE_RabbitV35 分钟前
【深度强化学习 DRL 快速实践】近端策略优化 (PPO)
算法
Wendy_robot1 小时前
【滑动窗口+哈希表/数组记录】Leetcode 438. 找到字符串中所有字母异位词
c++·算法·leetcode
程序员-King.1 小时前
day49—双指针+贪心—验证回文串(LeetCode-680)
算法·leetcode·贪心算法·双指针
转基因2 小时前
Codeforces Round 1020 (Div. 3)(题解ABCDEF)
数据结构·c++·算法
whoarethenext2 小时前
c++的jsoncpp使用
开发语言·c++·jsoncpp
我想进大厂3 小时前
图论---Kruskal(稀疏图)
数据结构·c++·算法·图论
@Aurora.3 小时前
数据结构手撕--【二叉树】
数据结构·算法
victd3 小时前
什么是AutoRec?
算法