LeetCode-Closest Binary Search Tree Value II

Posted LiBlog

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode-Closest Binary Search Tree Value II相关的知识,希望对你有一定的参考价值。

Given a non-empty binary search tree and a target value, find k values in the BST that are closest to the target.

Note:

  • Given target value is a floating point.
  • You may assume k is always valid, that is: k ≤ total nodes.
  • You are guaranteed to have only one unique set of k values in the BST that are closest to the target.

Follow up:
Assume that the BST is balanced, could you solve it in less than O(n) runtime (where n = total nodes)?

Analysis:

Use inorder traverse, put all predecessors into a stack, for every successor, put all pres that has smaller gap than that successor into resList and then put this successor into resList.

Solution:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<Integer> closestKValues(TreeNode root, double target, int k) {
        Stack<Integer> pres = new Stack<Integer>();
        LinkedList<Integer> resList = new LinkedList<Integer>();
        closestKValuesRecur(root,target,k,pres,resList);
        // If not enough in resList, put more pres into resList. This is because successor is too little.
        while (resList.size()<k && !pres.empty()){
                resList.addFirst(pres.pop());
        }
        return resList;        
    }

    public void closestKValuesRecur(TreeNode curNode, double target, int k, Stack<Integer> pres, LinkedList<Integer> resList){
        if (curNode == null) return;
        if (resList.size()==k) return;

        // inorder traverse.
        closestKValuesRecur(curNode.left,target,k,pres,resList);

        // check curNode
        if (curNode.val >= target){
            while (resList.size()<k && !pres.empty() && target-pres.peek() < curNode.val-target){
                resList.addFirst(pres.pop());
            }
            if (resList.size()<k){
                resList.addLast(curNode.val);
            } else {
                return;
            }
        } else {
            pres.push(curNode.val);
        }

        closestKValuesRecur(curNode.right,target,k,pres,resList);
    }
}

 

以上是关于LeetCode-Closest Binary Search Tree Value II的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode Count Binary Substrings

[Leetcode] Binary tree-- 572. Subtree of Another Tree

nowcoder3274D binary

Leetcode 之Binary Tree Preorder Traversal(42)

cf 1370E. Binary Subsequence Rotation (转化)

696. Count Binary Substrings