C++刷Leetcode 653.两数之和IV

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

案例 1:

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

Target = 9

输出: True
 

案例 2:

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

Target = 28

输出: False

来源https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst/

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool findTarget(TreeNode* root, int k) {
        vector<int> ivec;
        if(!root)
            return false;
        TreeNode *p = root;
        stack<TreeNode *> s;
        while(!s.empty() || p)
        {
            while(p)
            {
                s.push(p);
                p = p -> left;
            }
            if(!s.empty())
            {
                ivec.push_back(s.top() -> val);
                p = s.top() -> right;
                s.pop();
            }
        }
        unordered_map<int, int> hash;
        for(int i = 0; i != ivec.size(); ++i)
        {
            if(hash.count(k - ivec[i]))
                return true;
            hash[ivec[i]] = i;
        }
        return false;
    }
};