1. 问题描述
给定一棵二叉树和两个节点p、q,查找这两个节点最近的公共父节点。
2. 方法与思路
由于给定的是一颗普通的二叉树,与之前的二叉搜索树不一样,子树之间没有大小关系。但是可以用另一种思路,当前节点如果是p,q的公共最先节点,则p,q一定在当前节点的左右子树中。可以用递归解决。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL || root == p || root == q) return root;
TreeNode *ltree = lowestCommonAncestor(root->left,p,q);
TreeNode *rtree = lowestCommonAncestor(root->right,p,q);
if(ltree && rtree) return root;
return ltree?ltree:rtree;
}
};
其实这是一个经典的LCA问题,网上有好多资料可供参考。分为离线算法(Tarjan)和在线算法(RMQ)。