zoukankan      html  css  js  c++  java
  • leetcode 236. 二叉树的最近公共祖先

    题目描述:

    给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

    百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

    例如,给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]

     

    示例 1:

    输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
    输出: 3
    解释: 节点 5 和节点 1 的最近公共祖先是节点 3。

    示例 2:

    输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
    输出: 5
    解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。

    说明:

    • 所有节点的值都是唯一的。
    • p、q 为不同节点且均存在于给定的二叉树中。

    思路分析:

    这道题一开始想得比较复杂,对于p、q两个结点,都分别用两个双向队列去树中从根往下查找这两个结点,并将其对应的路径存在队列中。最后再从队首开始做出队操作,知道两个队列的队首不相等,那么最后一个相等的结点即为最近的公共祖先。这个思路可以,但在实现的过程中其实还是比较繁琐的,在记录路径的过程中其实还涉及到了回溯的操作。

    换一个思路,与树相关的问题,其实用递归来解决会更简单。这里我的问题实际上是对于一个问题,没有首先做分析思考,最直接的去想做法。其实分析起来,包括三类情况,p、q分别位于当前根结点的左右子树中,那么当前的根结点即为最近公共祖先;p、q都位于左子树中,那么当前的根结点就换为左子树,递归调用这个找公共祖先的函数;右子树的情况是类似的。所以实际只需要写两个函数,一个函数是判断结点是否在当前树中,一个是递归的找最近公共祖先。那么找公共祖先的这个函数的递归停止条件就是当前的根为p或q任意结点,或是p和q分别在当前根结点的左右子树中。 

    代码:

     1 /**
     2  * Definition for a binary tree node.
     3  * struct TreeNode {
     4  *     int val;
     5  *     TreeNode *left;
     6  *     TreeNode *right;
     7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     8  * };
     9  */
    10 class Solution {
    11 public:
    12     bool in_tree(TreeNode* root, TreeNode* cur)
    13     {
    14         if(root == NULL)
    15             return false;
    16         if(root == cur)
    17             return true;
    18         else if(in_tree(root->left, cur) || in_tree(root->right, cur))
    19             return true;
    20         else
    21             return false;
    22     }
    23     TreeNode* lowestCommonAncestorCore(TreeNode* root, TreeNode* p, TreeNode* q)
    24     {
    25         if(p == root || q == root)
    26             return root;
    27         if(in_tree(root->left, p) && in_tree(root->right, q))
    28             return root;
    29         else if(in_tree(root->left, q) && in_tree(root->right, p))
    30             return root;
    31         else if(in_tree(root->left, p) && in_tree(root->left, q))
    32             return lowestCommonAncestorCore(root->left, p, q);
    33         else
    34             return lowestCommonAncestorCore(root->right, p, q);
    35     }
    36     TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
    37         if(root == NULL || p == NULL || q == NULL)
    38             return root;
    39         
    40         TreeNode* res = lowestCommonAncestorCore(root, p, q);
    41         return res;
    42     }
    43 };
  • 相关阅读:
    SSH学习-struts2整合spring报错'Could not open ServletContext resource [/WEB-INF/applicationContext.xml]'
    YAML学习
    配置composer代理
    Windows下配置PHPUnit(pear已弃用,使用phpunit.phar)
    算法-第四版-练习1.3.9解答
    算法-第四版-练习1.3.10解答
    算法-第四版-练习1.3.11解答
    算法-第四版-练习1.3.12解答
    算法-第四版-练习1.3.13解答
    算法-第四版-练习1.3.14解答
  • 原文地址:https://www.cnblogs.com/LJ-LJ/p/11217873.html
Copyright © 2011-2022 走看看