235. 二叉搜索树的最近公共祖先(leetcode/C++)

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

百度百科中最近公共祖先的定义为: “对于有根树T的两个结点u、v,最近公共祖先表示一个结点x,满足x是u、v的祖先且x的深度尽可能大。”(一个节点也可以是它自己的祖先

例如,给定二叉搜索树:  root = [6,2,8,0,4,7,9,null,null,3,5]

        _______6______
       /              \
    ___2__          ___8__
   /      \        /      \
   0      _4       7       9
         /  \
         3   5

示例 1:

输入: root, p = 2, q = 8
输出: 6 
解释: 节点 2 和节点 8 的最近公共祖先是 6。

示例 2:

输入: root, p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为指定节点自身。


思想:由于是基于二叉树搜索树进行查找,那么问题简化了很多。这里我用到了两个vector分别记录了两个结点的搜索路径,然后再遍历两个vector找出最后一个相同的结点,这里需要注意的是题目要求返回结点指针,那么在定义vector存储变量类型的时候需要注意了。细节见代码:
/**
 * 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) {
        vector left;
        vector right;
        TreeNode *pp = root;
        TreeNode *qq = root;
        while (pp != NULL)
        {
            left.push_back(pp);
            if (p->val < pp->val)
            {
                pp = pp->left;
                
            }
            else if (p->val > pp->val)
            {
                pp = pp->right;
            }
            else
            {
                break;
            }
        }
        while (qq != NULL)
        {
            right.push_back(qq);
            if (q->val < qq->val)
            {
                qq = qq->left;
                
            }
            else if (q->val > qq->val)
            {
                qq = qq->right;
            }
            else
            {
                break;
            }
        }
        
        int i = 0;
        int j = 0;
        TreeNode *temp;
        
        while (i < left.size() && j < right.size())
        {
            if (left[i] == right[j])
            {
                temp = left[i];
            }
            i++;
            j++;
        }
        
        return temp;
    }
};




你可能感兴趣的:(leetcode,C++,树)