剑指 Offer II 056. 二叉搜索树中两个节点之和

给定一个二叉搜索树的 根节点 root 和一个整数 k , 请判断该二叉搜索树中是否存在两个节点它们的值之和等于 k 。假设二叉搜索树中节点的值均唯一。

示例 1：

输入: root = [8,6,10,5,7,9,11], k = 12
输出: true
解释: 节点 5 和节点 7 之和等于 12

示例 2：

输入: root = [8,6,10,5,7,9,11], k = 22
输出: false
解释: 不存在两个节点值之和为 22 的节点

提示：

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

---

递归：

class Solution {
private:
    unordered_set<int> hash;
    bool find(TreeNode* root, int k){
        if(root == nullptr)
            return false;
        if(hash.count(k-root->val))
            return true;
        else
            hash.emplace(root->val);
        int left = find(root->left,k);
        int right = find(root->right,k);
        return left==true?left:right;
    }
public:
    bool findTarget(TreeNode* root, int k) {
        return find(root,k);
    }
};

迭代：

class Solution {
public:
    bool findTarget(TreeNode* root, int k) {
        if(root == nullptr)
            return false;
        unordered_set<int> hash;
        stack<TreeNode*> stk;
        TreeNode* cur=root;
        while(cur!=nullptr||!stk.empty())
        {
            while(cur!=nullptr)
            {
                stk.push(cur);
                cur = cur->left;
            }
            cur = stk.top();
            stk.pop();
            if(hash.count(k-cur->val))
                return true;
            hash.emplace(cur->val);
            cur = cur->right;
        }
        return false;
    }
};