236. 二叉树的最近公共祖先

复制代码

https://leetcode.cn/problems/lowest-common-ancestor-of-a-binary-tree/?envType=study-plan-v2&envId=top-interview-150

复制代码

思路：因为自己可以是自己的祖先，所以这题其实是在问哪棵子树（最小）可以同时将两个节点包含，我们需要返回该子树的根节点；因为没有重复数所以我们可以用cnt来记录以当前节点为根的子树包含这两个节点的数量，我们可以用cnt=当前节点本身是否相等+左右子树包含的数量, 就可以算出当前节点为根的子树包含这两个节点的数量，当第一次等于2的时候就代表我们找到了。

java 复制代码

public class Main {
    int cnt = 0;
    TreeNode goal = null;
    boolean flag = false;
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        method(root, p, q);
        return goal;
    }

    public int method(TreeNode node, TreeNode p, TreeNode q) {
        if (node == null) {
            return 0;
        }
        //当前节点为根的子树包含这两个节点的数量
        int cnt = ((node.val == p.val || node.val == q.val) ? 1 : 0) + method(node.left, p, q) + method(node.right, p, q);
        // 第一次等于2，我们记录一下根节点，后续不再进入该部分
        if (cnt == 2 && !flag) {
            goal = node;
            flag = true;
        }
        return cnt;
    }
}