272 最接近的二叉搜索树值 II

题目描述：
给定一个不为空的二叉搜索树和一个目标值 target，请在该二叉搜索树中找到最接近目标值 target 的 k 个值。

注意：
给定的目标值 target 是一个浮点数
你可以默认 k 值永远是有效的，即 k ≤ 总结点数
题目保证该二叉搜索树中只会存在一种 k 个值集合最接近目标值

示例：
输入: root = [4,2,5,1,3]，目标值 = 3.714286，且 k = 2

输出: [4,3]
拓展：
假设该二叉搜索树是平衡的，请问您是否能在小于 O(n)（n 为总结点数）的时间复杂度内解决该问题呢？

方法1：
主要思路：
（1）使用最小堆；
（2）将树中结点的值和给定的目标值计算出对应的差值，并和结点值一起组成pair类型，存储到最小堆中；
（3）再从最小堆中弹出前面的k个值对应的结点值，存储到结果中即可；

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,double& target,priority_queue<pair<double,int>,vector<pair<double,int>>,greater<pair<double,int>>>&q){
        if(root==NULL){//终止条件
            return ;
        }
        //存储结果，差值使用绝对值存储
        q.push(pair<double,int>(abs(((double)root->val)-target),root->val));
        //递归
        dfs(root->left,target,q);
        dfs(root->right,target,q);
    }
    vector<int> closestKValues(TreeNode* root, double target, int k) {
        priority_queue<pair<double,int>,vector<pair<double,int>>,greater<pair<double,int>>> q;//最小堆
        dfs(root,target,q);//深度搜索，获得最小堆
        vector<int> res;
        for(int i=0;i<k;++i){//将最小堆中的前 k 个元素取出
            res.push_back(q.top().second);
            q.pop();
        }
        return res;
    }
};

方法2：
主要思路：
（1）使用最大堆；
（2）使用最大堆存储 k 个值，超过 k 个值时，弹出多出的元素，最后将最大堆中的 k 个元素的值输出给数组即可；

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,double& target,int&k,priority_queue<pair<double,int>,vector<pair<double,int>>,less<pair<double,int>>>&q){
        if(root==NULL){
            return ;
        }
        q.push(pair<double,int>(abs(((double)root->val)-target),root->val));
        if(q.size()>k){//保证最大堆中的数量满足要求
            q.pop();
        }
        dfs(root->left,target,k,q);
        dfs(root->right,target,k,q);
    }
    vector<int> closestKValues(TreeNode* root, double target, int k) {
        priority_queue<pair<double,int>,vector<pair<double,int>>,less<pair<double,int>>> q;//最大堆
        dfs(root,target,k,q);
        vector<int> res;
        while(!q.empty()){//将最大堆中的值存入到结果中
            res.push_back(q.top().second);
            q.pop();
        }
        return res;
    }
};

方法3：
主要思路：
（1）二分搜索；
（2）先使用中序遍历将树中的元素存储到辅助数组中，然后在辅助数组中使用二分搜索，找出需要的 k 个值的第一个值的位置，再将该范围的值直接输出成数组即可；

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
	//中序遍历，将树的结点有序点的存入到数组中
    void inorder(TreeNode* root,vector<int>& sort_val){
        if(root==NULL){
            return;
        }
        inorder(root->left,sort_val);
        sort_val.push_back(root->val);
        inorder(root->right,sort_val);
    }
    vector<int> closestKValues(TreeNode* root, double target, int k) {
        vector<int> sort_val;
        inorder(root,sort_val);//获得有序的数组
        //二分搜索的左右端点
        int left=0;
        int right=sort_val.size()-k;
        //二分搜索
        while(left<right){
            int mid=left+(right-left)/2;
            //找出给定的 k 个值的范围的 第一个值的位置
            if(target-sort_val[mid]>sort_val[mid+k]-target){
                left=mid+1;
            }
            else{
                right=mid;
            }
        }
        //返回结果的 k 个值
        return vector<int>(sort_val.begin()+left,sort_val.begin()+left+k);
    }
};