给定一个二叉树,找到该树中指定节点的最近公共祖先。
其中节点5和节点1的最近公共祖先是节点3。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉树中
解决思路:
1.如果根节点的左右子树分别存在p,q,则返回此刻的根节点。
2.如果根节点和左子树都不是节点,p,q,则可以根据公共祖先必定在右子树。
3.否则公共祖先必在则在左子树。
/** * 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) return NULL; if(root == p || root == q|| root==NULL){ return root; } TreeNode* left_node = lowestCommonAncestor(root->left, p, q); //cout << "left_node:" << left_node->val << ","; TreeNode* right_node = lowestCommonAncestor(root->right, p, q); //cout << "right_node" << right_node->val << ","; //if() if((left_node == p && right_node == q) || (right_node == p && left_node == q)) { return root; } else if(!left_node){ return right_node; } else{ return left_node; } //return NULL; } };