力扣2476.二叉搜索树最近节点查询
-
二叉搜索树 + 中序遍历 = 严格递增数组
- 在数组上做二分 找到第一个>=q的元素的下标j
- 若j < n 则a[j]为maxx
- 若j-1>0 && a[j] != p 则a[j-1]为minx
-
class Solution {vector<int> a;void dfs(TreeNode* node){if(node == nullptr) return;dfs(node->left);a.push_back(node->val);dfs(node->right);}public:vector<vector<int>> closestNodes(TreeNode* root, vector<int>& queries) {dfs(root);int n = a.size();vector<vector<int>> res;for(int q:queries){//找第一个>=q元素的下标int j = ranges::lower_bound(a,q) - a.begin();int maxx = j < n ? a[j] : -1;//若j==n 说明当前q为最大值 j-1一定是<q的下标if(j == n || a[j] != q) j --;int minx = j >= 0 ? a[j] : -1;res.push_back({minx,maxx});}return res;}};
