进攻LeedCode653 . 两数之和IV-输入BST

LeedCode653:

        给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true

 

 

提示:

二叉树的节点个数的范围是  [1, 104].
-104 <= Node.val <= 104
root 为二叉搜索树
-105 <= k <= 105

解法1:哈希表 + 树的搜索


        在递归搜索过程中记录下相应的节点值(使用 Set 集合),如果在遍历某个节点 xx 时发现集合中存在 k - x.valk−x.val,说明存在两个节点之和等于 kk,返回 True,若搜索完整棵树都没有则返回 False。        

        代码如下:

class Solution {
    Set<Integer> set = new HashSet<>();
    public boolean findTarget(TreeNode root, int k) {
        if (root == null) return false;
        if (set.contains(k - root.val)) return true;
        set.add(root.val);
        return findTarget(root.left, k) || findTarget(root.right, k);
    }
}

解法2:双指针 + BST 中序遍历


        解法一中没有利用 BST 特性,利用 BST 中序遍历有序的特性,我们可以实现类似「双指针」的效果。

起始先让 BST 的最左链和最右链完全入栈,此时栈顶元素为 BST 中的最小值和最大值,分别使用 l 和 r 充当指针代指,根据两指针指向的节点值之和与 kk 的大小关系来指导如何让 l 和 r 移动,l 的移动过程其实就是找下一个比 l.val 更大的值,而 r 的移动过程其实就是找下一个比 r.val 更小的值。

        代码如下:

class Solution {
    public boolean findTarget(TreeNode root, int k) {
        Deque<TreeNode> ld = new ArrayDeque<>(), rd = new ArrayDeque<>();
        TreeNode temp = root;
        while (temp != null) {
            ld.addLast(temp);
            temp = temp.left;
        }
        temp = root;
        while (temp != null) {
            rd.addLast(temp);
            temp = temp.right;
        }
        TreeNode l = ld.peekLast(), r = rd.peekLast();
        while (l.val < r.val) {
            int t = l.val + r.val;
            if (t == k) return true;
            if (t < k) l = getNext(ld, true);
            else r = getNext(rd, false);
        }
        return false;
    }
    TreeNode getNext(Deque<TreeNode> d, boolean isLeft) {
        TreeNode node = isLeft ? d.pollLast().right : d.pollLast().left;
        while (node != null) {
            d.addLast(node);
            node = isLeft ? node.left : node.right;
        }
        return d.peekLast();
    }
}

        有任何问题,随时联系博主解决,感谢阅读,谢谢谢谢谢谢~~~

猜你喜欢

转载自blog.csdn.net/chen_kai_fa/article/details/123630431
今日推荐