No.127 - LeetCode1214

ShellDawn

于 2019-10-05 23:31:35 发布

阅读量81

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/102181804

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种算法，用于在两个二叉搜索树中查找是否存在两个节点，其值相加等于给定的目标值。通过深度优先搜索(DFS)遍历一棵树并记录节点值，再遍历另一棵树检查是否存在目标值减去当前节点值的配对。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    void dfs(TreeNode* now,map<int,int>& mp){
        mp[now->val] = 1;
        if(now->left != NULL) dfs(now->left,mp);
        if(now->right != NULL) dfs(now->right,mp);
    }
    bool dfsB(TreeNode* now,int& target,map<int,int>& mp){
        if(mp[target-now->val] == 1) return true;
        bool ans = false;
        if(now->left != NULL) ans = ans || dfsB(now->left,target,mp);
        if(now->right != NULL) ans = ans || dfsB(now->right,target,mp);
        return ans;
    }
    bool twoSumBSTs(TreeNode* root1, TreeNode* root2, int target) {
        if(root1 == NULL) return false;
        if(root2 == NULL) return false;
        map<int,int> mp;
        dfs(root1,mp);
        return dfsB(root2,target,mp);
    }
};