6.15 二叉树的最近公共祖先

二叉树的最近公共祖先

注意：最近公共祖先可以是节点本身

      public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
          if(root==null)return null;
          //如果查找的节点和p或者q相等，也没有必要再往下查找，而是返回去找他们的父节点
          //最近公共祖先节点可以为节点本身。
          if(p==root || q==root)return root;
          //这递归函数在查找以root为根节点的二叉树中p、q结点的父节点
          //拿到左子树、右子树p、q的根节点
          TreeNode left=lowestCommonAncestor(root.left, p, q);
          TreeNode right=lowestCommonAncestor(root.right, p, q);
          if(left==null && right==null)return null;
          else if(left!=null && right!=null)return root;
          else return left==null ? right:left;
      }

 下面这版代码更好理解

      public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
          if(root==null)return null;
          //如果查找的节点和p或者q相等，也没有必要再往下查找，而是返回去找他们的父节点
          //最近公共祖先节点可以为节点本身。
          if(p==root || q==root)return root;
          //这递归函数在查找以root为根节点的二叉树中p、q结点的父节点
          //拿到左子树、右子树p、q的根节点
          TreeNode left=lowestCommonAncestor(root.left, p, q);
          TreeNode right=lowestCommonAncestor(root.right, p, q);
          //都在右子树
          if(left==null)return right;
          //都在左子树
          else if(right==null)return left;
          //两边各一个的情况
          return root;
      }

 

二叉搜索树的最近公共节点

思想：这个每层的情况可以根据p、q在左子树还是右子树来讨论

那么就有三种情况，都在左子树，都在右子树，两边各一个

 public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
          if(root==null || p==null || q==null)return root;
//          TreeNode left=lowestCommonAncestor(root.left, p, q);
//          TreeNode right=lowestCommonAncestor(root.right, p, q);
          if(root.val>p.val && root.val<q.val)return root;//在两边，其实可以删掉这种情况，因为除了后面两种就只剩这一种了，可以直接return
          //p、q都在左子树
          else if(root.val>p.val && root.val>q.val)return lowestCommonAncestor(root.left, p, q);
          //都在右子树
          else if(root.val<p.val && root.val<q.val)return lowestCommonAncestor(root.right, p, q);
          return root;
      }

总结：公共祖先的问题可以归类为p、q节点是否在同一子树中的问题，也就是三种情况。