LeetCode 653 两数之和IV-输入BST[map 二叉树] HERODING的LeetCode之路

Posted 2022-03-23 HERODING23

解题思路：
用map存储每次遍历到的数，并且查找map中是否存在k减去该数的数，如果有，则说明存在目标两数，返回true，否则返回false，代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode 
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) 
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) 
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) 
 * ;
 */
class Solution 
private:
    unordered_map<int, int> mp;
public:
    bool findTarget(TreeNode* root, int k) 
        if(root == nullptr) return false;
        if(mp.count(k - root->val)) return true;
        mp[root->val] ++;
        return findTarget(root->left, k) || findTarget(root->right, k);
    
;