653. 两数之和 IV - 输入 BST(Java实现)

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

案例 1:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 9

输出: True

案例 2:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 28

输出: False
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
//利用中序遍历,定义HashSet,每次加入的值为k-root.val,当在遇到相等的,就说明找到
class Solution {
    public boolean findTarget(TreeNode root, int k) {
        if(root==null)return false;
       Stack<TreeNode> stack=new Stack<>();
        HashSet<Integer> set=new HashSet<>();
        while(root!=null || !stack.isEmpty()){
            while(root!=null){
                stack.push(root);
                root=root.left;
            }
            root=stack.pop();
            if(set.contains(root.val)) 
                return true;
            else 
                set.add(k-root.val);
            root=root.right;
           }
        return false;
        
    }
}

猜你喜欢

转载自blog.csdn.net/ccccc1997/article/details/81460270
今日推荐