文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給定一個不為空的二叉搜索樹和一個目標(biāo)值 target，請在該二叉搜索樹中找到最接近目標(biāo)值 target 的 k 個值。

注意：
給定的目標(biāo)值 target 是一個浮點數(shù)
你可以默認(rèn) k 值永遠是有效的，即 k ≤ 總結(jié)點數(shù)
題目保證該二叉搜索樹中只會存在一種 k 個值集合最接近目標(biāo)值

示例： 輸入: root = [4,2,5,1,3]，目標(biāo)值 = 3.714286，且 k = 24/ \2 5/ \ 1 3輸出: [4,3]

拓展：
假設(shè)該二叉搜索樹是平衡的，請問您是否能在小于 O(n)（n 為總結(jié)點數(shù)）的時間復(fù)雜度內(nèi)解決該問題呢？

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/closest-binary-search-tree-value-ii
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

類似題目：LeetCode 658. 找到 K 個最接近的元素（二分查找）

• 使用stack，中序遍歷bst，是有序的
• 將差值最小的k個元素的<差值，自身值>插入優(yōu)先隊列
• 隊列滿了k個，且差值為正，且大于堆頂，可以提前結(jié)束

struct cmp {bool operator()(vector<double>& a, vector<double>& b){return a[0] < b[0];//差值大的在上} }; class Solution { public:vector<int> closestKValues(TreeNode* root, double target, int k) {stack<TreeNode*> s;priority_queue<vector<double>, vector<vector<double>>, cmp> q;TreeNode* cur;while(root || !s.empty()){while(root){s.push(root);root = root->left;}cur = s.top();s.pop();root = cur->right;if(q.size() < k)q.push({fabs(cur->val-target), double(cur->val)});else if(q.size()==k && q.top()[0] > fabs(cur->val-target)){ //有更小的差值q.pop();q.push({fabs(cur->val-target), double(cur->val)});}if(q.size()==k && cur->val-target >= q.top()[0])break;}vector<int> ans(k);int i = 0;while(!q.empty()){ans[i++] = q.top()[1];q.pop();}return ans;} };

52 ms 22.4 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關(guān)注我的公眾號（Michael阿明），一起加油、一起學(xué)習(xí)進步！

總結(jié)

以上是生活随笔為你收集整理的LeetCode 272. 最接近的二叉搜索树值 II（栈+优先队列）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。