编程题-两数之和&输入二叉搜索树-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_43287713/article/details/144974314

题目：

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

解法一（深度优先搜索+哈希表）

使用深度优先搜索的方式遍历整棵树，用哈希表记录遍历过的节点的值。对于一个值为x的节点，我们检查哈希表中是否存在k-x即可。如果存在对应的元素，我们就可以在该树上找到两个节点的和为k；否则我们将x放入到哈希表中，如果遍历完整棵树都不存在对应的元素，那么该树上不存在两个和为k的节点。如下代码：

class Solution {
public:
    //定义哈希表
    unordered_set<int> hashTable;
    //查找是否存在指定的k-root->val数，如果没有则insert加入到哈希表中，并递归(深度搜索继续遍历左右子树)
    bool findTarget(TreeNode *root, int k) {
        if (root == nullptr) {
            return false;
        }
        if (hashTable.count(k - root->val)) {
            return true;
        }
        hashTable.insert(root->val);
        //递归调用findTarget函数，遍历root的左右子树
        return findTarget(root->left, k) || findTarget(root->right, k);
    }
};

解法二（广度优先搜索+哈希表）

创建一个哈希表和一个队列，将根节点加入队列中，然后执行以下操作，从队列中取出队头，假设其值为x，检查哈希表中是否存在k-x，如果存在，返回True，否则将该节点的左右的非空子节点加入队尾，重复以上操作直到队列为空，如果队列为空，说明树上不存在两个和为k的节点，返回False。如下为笔者代码：

class Solution {
public:
    bool findTarget(TreeNode* root, int k) {
        queue<TreeNode*> queue1;
        unordered_map<int,int> hashTable1;
        if(root==nullptr){
            return false;
        }
        else{
            queue1.push(root);
            hashTable1[k-(root->val)]=root->val;
        }
        while(!queue1.empty()){
            if(queue1.front()->left!=nullptr){
                if(hashTable1.find(queue1.front()->left->val) != hashTable1.end()){
                    return true;
                }
                else{
                    hashTable1[k-(queue1.front()->left->val)]=queue1.front()->left->val;
                    queue1.push(queue1.front()->left);
                }
                
            }
            if(queue1.front()->right!=nullptr){
                if(hashTable1.find(queue1.front()->right->val) != hashTable1.end()){
                    return true;
                }
                else{
                    hashTable1[k-(queue1.front()->right->val)]=queue1.front()->right->val;
                    queue1.push(queue1.front()->right);
                }
            }
            queue1.pop();
        }
        return false;
    }
};

笔者小记：关于哈希表和队列的基本操作笔者进行了一些总结，基本操作需熟练掌握

1、哈希表std::unordered_map hashTable;哈希表的查找和删除均是键元素，而值元素则是在找到键元素的基础上进行操作的。哈希表不能直接对值进行操作（例如直接查找值或直接删除值）

创建哈希表unordered_map<int, int> hashTable1;第一个int为“键”，第二个int为“值”

插入元素：hashTable[键]=值

查找元素：hashTable.count()，hashTable.find()均是查找键

删除元素：hashTable.erase()删除键

2、队列std::queue；队列区别与栈，有一些操作符号是和栈相同的，有些是不同的，具体来说

入列/入栈：queue.push()，stack.push()，操作符相同

出列/出栈：queue.pop()，stack.pop()，操作符相同

取列顶元素/取栈顶元素：queue.front()，stack.top()，操作符不同

取队列尾部元素：queue.back()，栈不支持直接访问栈最底部元素

判断队列/栈是否为空：queue.empty()，stack.empty()，操作符相同