Leetcode 235. 二叉搜索树的最近公共祖先 C++

Leetcode 235. 二叉搜索树的最近公共祖先 C++_第1张图片

思路

因为是二叉搜索树,可以充分利用二叉搜索树的性质来做这道题。二叉搜索树左子树的节点值一定全部小于根节点,右子树的节点值全部大于根节点。所以:
(1)如果所找的两个节点一个大于根节点的值,一个小于根节点的值,那么这个根节点就为最近的祖先。
(2)若是同时小于根节点,那么就去左子树中找。
(3)如果同时大于根节点,就去右子树中找答案。


最近自己编写递归程序的能力有下滑,带返回值递归的处理居然都不会了。递归的时候,一样用return就好

解答

/**
 * 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) return root;
        if( (root->val >= p->val && root->val <= q->val) || (root->val <= p->val && root->val >= q->val)) return root;
        else if(root->val >p->val && root->val >q->val) return lowestCommonAncestor(root->left, p, q);
        else if(root->val <p->val && root->val <q->val) return lowestCommonAncestor(root->right, p, q);
        else return NULL;
    }
};

你可能感兴趣的:(leetcode)