二叉树的最近祖先

给定一个二叉树,找到该树中指定节点的最近公共祖先。

其中节点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;
        
    }
};

 

posted @ 2019-09-09 22:19  卷积  阅读(476)  评论(0编辑  收藏  举报