题解 | #牛群的数量匹配# dfs+集合

牛群的数量匹配

https://www.nowcoder.com/practice/cab27b84774d40a5aaea2ee69e2d0b77

使用dfs遍历树,将遍历结果加入集合中,如果集合中存在与当前节点值之和为目标值的元素,返回true

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  private:
    bool dfs(TreeNode* root, unordered_set<int>& st, int k) {
        if (!root) {
            return false;
        }
        if (st.find(k - root->val) != st.end()) {
            return true;
        }
        st.insert(root->val);
        return dfs(root->left, st, k) || dfs(root->right, st, k);
    }
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @param k int整型
     * @return bool布尔型
     */
    bool findTarget(TreeNode* root, int k) {
        unordered_set<int> st;
        return dfs(root, st, k);
    }
};

时间复杂度:在最坏情况下,需要遍历整棵二叉搜索树,时间复杂度为 O(n),其中 n 为树中节点的数量。

空间复杂度:

使用了一个整型无序集合 st 来存储节点值,最坏情况下需要存储所有节点的值,空间复杂度为 O(n);

递归调用时,需要考虑函数调用栈的空间,最坏情况下递归深度为树的高度,空间复杂度为 O(h),其中 h 为树的高度。

全部评论

相关推荐

点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-04 14:23
steelhead:你回的有问题,让人感觉你就是来学习的
点赞 评论 收藏
分享
06-08 22:25
门头沟学院 Java
从零开始的转码生活:这hr不会打开手机不分青红皂白给所有人群发这句话,过一会再给所有人再发一遍,这肯定会有重复的,不管,再过一会再发一遍
点赞 评论 收藏
分享
05-20 21:57
已编辑
门头沟学院 Java
喜欢吃卤蛋的悲伤蛙在...:建信融通没消息吧,我2说有实习挂简历不理了
点赞 评论 收藏
分享
完美的潜伏者许愿简历...:隐藏信息被你提取出来了,暗示,这就是暗示
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务