其实树的很多题,都是遍历的套路,先处理左子树,右子树还是先处理根节点引出不同种类的遍历类型。
层序遍历也存在着一些套路,我目前接触过的层序遍历的套路都只是一个类型,所以暂时按下不表,只讨论后序,前序,中序遍历。
236. 二叉树最近公共祖先 medium
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
输入描述:
第一行输入两个整数 n 和 root,n 表示二叉树的总节点个数,root 表示二叉树的根节点。 以下 n 行每行三个整数 fa,lch,rch,表示 fa 的左儿子为 lch,右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子,rch同理) 最后一行为节点 o1 和 o2。
分析:这题是二叉搜索树的拓展
但是因为无法像二叉搜索树那样直接通过根节点判断两个节点是否处于根节点的左边和右边,所以得进行类似后序遍历的递归形式。先处理root的两棵子树,判断root->left是否存在节点,root->right是否存在节点,再判断root情况。
代码:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(!root || root == p || root == q) return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if(left && right) return root;
return (left) ? left : right;
}
待续