【Leetcode】272. Closest Binary Search Tree Value II-CSDN博客

本文链接：https://blog.csdn.net/qq_46105170/article/details/127787022

该博客介绍了如何在给定的二叉搜索树中找到最接近目标值的k个数。算法使用双端队列进行中序遍历，存储小于等于目标值的最大k个数和大于目标值的最小k个数，然后通过双指针扩展至k个结果。时间复杂度为O(n+k)，空间复杂度为O(h)。

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

题目地址：

https://leetcode.com/problems/closest-binary-search-tree-value-ii/description/

给定一棵 $n$ 个节点的二叉搜索树，再给定一个值 $t$ 和一个正整数 $k$ ，求树中离 $t$ 最近的 $k$ 个数。题目保证 $k\le n$ 。

带着两个双端队列对BST进行中序遍历，将小于等于 $t$ 的最大的 $k$ 个数和大于 $t$ 最小的 $k$ 个数存下来（如果不够 $k$ 个则全存）。最后用类似双指针的办法，从离 $t$ 最近的数开始两边扩展至 $k$ 个即可。代码如下：

class Solution {
 public:
  vector<int> closestKValues(TreeNode* root, double t, int k) {
    deque<int> dq1, dq2;
    dfs(root, t, k, dq1, dq2);
    vector<int> res;
    while (res.size() < k) {
      if (dq2.empty() ||
          dq1.size() && abs(dq1.back() - t) <= abs(dq2.front() - t)) {
        res.push_back(dq1.back());
        dq1.pop_back();
      } else {
        res.push_back(dq2.front());
        dq2.pop_front();
      }
    }

    return res;
  }

  void dfs(TreeNode* root, double t, int k, deque<int>& dq1, deque<int>& dq2) {
    if (!root) return;
    dfs(root->left, t, k, dq1, dq2);
    if (root->val <= t) {
      dq1.push_back(root->val);
      if (dq1.size() > k) dq1.pop_front();
    } else if (dq2.size() < k)
      dq2.push_back(root->val);

    dfs(root->right, t, k, dq1, dq2);
  }
};