2021-09-03

最新推荐文章于 2023-11-21 10:53:33 发布

原创最新推荐文章于 2023-11-21 10:53:33 发布 · 138 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#java #数据结构 #二叉树

该博客介绍了如何在二叉搜索树中找到最接近目标值的K个节点。通过中序遍历，找到最接近目标的节点，然后从这个节点向左右两侧扩展，构建出答案列表。算法主要涉及到二叉搜索树的遍历和数值比较。

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

这题先中序遍历得到list，在遍历的时候顺便找到最接近于target的点，之后由这一点出发向左向右找离target最小的点加入到answer的list中输出。

class Solution {
    public List<Integer> closestKValues(TreeNode root, double target, int k) {
        List<Integer> list=new ArrayList<>();
        Stack<TreeNode> stack=new Stack<>();
        double number=Double.MAX_VALUE;
        int no=0;
        while(!stack.isEmpty()||root!=null){
            while(root!=null){
                stack.push(root);
                root=root.left;
            }
            root=stack.pop();
            list.add(root.val);
            if(Math.abs(root.val-target)<number){
                no=list.size()-1;
                number=Math.abs(root.val-target);
            }
            root=root.right;
        }
        List<Integer> answer=new ArrayList<>();
        answer.add(list.get(no));
        int i=no-1;
        int j=no+1;
        int size=list.size();
        while(answer.size()<k){
            if(i>=0&&j<size){
                if(Math.abs(list.get(i)-target)<=Math.abs(list.get(j)-target)){
                    answer.add(list.get(i--));
                }else{
                    answer.add(list.get(j++));
                }
            }
            if(i<0){
                for(int t=0;t<k-answer.size();){
                    answer.add(list.get(j++));
                }
            }
            if(j>=size){
                for(int t=0;t<k-answer.size();){
                    answer.add(list.get(i--));
                }
            }
        }
        return answer;
    }
}