【力扣-数据结构】【第 14 天】653. 两数之和 IV - 输入 BST

标题	653. 两数之和 IV - 输入 BST
难度	简单
天数	第14天,第2/3题
数据结构	树

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

示例 1：

输入: root = [5,3,6,2,4,null,7], k = 9
输出: true

示例 2：

输入: root = [5,3,6,2,4,null,7], k = 28
输出: false

 

提示:

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

以上内容来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：

1. 遍历二叉树，使用map存储每个节点的值，

	HashMap<Integer,Integer> map = new HashMap();

2. 用堆，来储存遍历每一个节点

	Stack<TreeNode> stack = new Stack();

3. 遍历过程中，先判断map是否包含k-node.val，如果包含那么就等于给定的目标结果k

	TreeNode node = stack.pop();
		int value = k - node.val;
		if(map.containsKey(value)){
			//如果差值存在于map中，代表存在两个数和=k
			return true;
		}
	}

4. 不相等，就添加进map

 	//如果map里不存在，就将当前值存入map
	map.put(node.val,node.val);

5. 然后将左右节点添加进栈

	//左子树存在，左子树存入栈中
	if(node.right != null){
		stack.push(node.right);
	}
	//右子树存在，右子树存入栈中
	if(node.left != null){
		stack.push(node.left);
	}

完整代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    //数据结构 第 14 天 2/3  树
    public boolean findTarget(TreeNode root, int k) {
        HashMap<Integer,Integer> map = new HashMap();
        Stack<TreeNode> stack = new Stack();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode node = stack.pop();
            int value = k - node.val;
            if(map.containsKey(value)){
                //如果差值存在于map中，代表存在两个数和=k
                return true;
            }else{
                //如果map里不存在，就将当前值存入map
                map.put(node.val,node.val);
            }
            //左子树存在，左子树存入栈中
            if(node.right != null){
                stack.push(node.right);
            }
             //右子树存在，右子树存入栈中
            if(node.left != null){
                stack.push(node.left);
            }
        }
        
        return false;
    }
}