Update 0236.二叉树的最近公共祖先.md

更新 0236题 Java版本代码

problems/0236.二叉树的最近公共祖先.md

@@ -221,42 +221,30 @@ public:
 
 
 ## Java 
+
 ```Java
 class Solution {
     public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
-        return lowestCommonAncestor1(root, p, q);
-    }
-    public TreeNode lowestCommonAncestor1(TreeNode root, TreeNode p, TreeNode q) {
-        if (root == null || root == p || root == q) {
+        if (root == null || root == p || root == q) { // 递归结束条件
             return root;
         }
-        TreeNode left = lowestCommonAncestor1(root.left, p, q);
-        TreeNode right = lowestCommonAncestor1(root.right, p, q);
-        if (left != null && right != null) {// 左右子树分别找到了，说明此时的root就是要求的结果
-            return root;
-        }
-        if (left == null) {
+
+        // 后序遍历
+        TreeNode left = lowestCommonAncestor(root.left, p, q);
+        TreeNode right = lowestCommonAncestor(root.right, p, q);
+
+        if(left == null && right == null) { // 若未找到节点 p 或 q
+            return null;
+        }else if(left == null && right != null) { // 若找到一个节点
             return right;
+        }else if(left != null && right == null) { // 若找到一个节点
+            return left;
+        }else { // 若找到两个节点
+            return root;
         }
-        return left;
     }
 }
-```
 
-
-```java
-// 代码精简版
-class Solution {
-    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
-        if (root == null || root.val == p.val ||root.val == q.val) return root;
-        TreeNode left = lowestCommonAncestor(root.left,p,q);
-        TreeNode right = lowestCommonAncestor(root.right,p,q);
-        if (left != null && right != null) return root;
-        else if (left == null && right != null) return right;
-        else if (left != null && right == null) return left;
-        else return null;
-    }
-}
 ```
 
 ## Python