235. 二叉搜索树的最近公共祖先
知识点:二叉树、回溯
时间:2020年9月27日
题目链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree/
题目
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
说明:
叶子节点是指没有子节点的节点。
示例1
输入:
root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出:
6
解释:
节点 2 和节点 8 的最近公共祖先是 6。
示例2
输入:
root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出:
2
解释:
节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。
说明:
所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉搜索树中。
解法
- 考虑到是二叉搜索树,我们可以找到从根节点到目标节点的路径,放入数组中
- 比较路径数组中的值,最后一个相同的值为答案的解
- 上面遍历了两遍树,但是我们可以根据二叉树的性质把两个节点一起遍历
- 如果两个节点都比此节点小,往这个节点的左子树遍历
- 如果两个节点都比此节点大,往这个节点的右子树遍历
- 如果一个比此节点大,一个小,这个节点就是最近公共祖先
代码
#include <stdio.h>
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
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<int> path1 = findpath(root, p);
vector<int> path2 = findpath(root, q);
int ans=0;
for(int i=0;i<path1.size()&&i<path2.size();i++)
{
if(path1[i]==path2[i])
ans = path1[i];
}
TreeNode node(ans);
return &node;
}
vector<int> findpath(TreeNode* root,TreeNode* taget){
vector<int> path;
TreeNode* tmp = root;
while(tmp != taget){
path.push_back(tmp->val);
if(tmp->val < taget->val)
tmp = tmp->right;
else if(tmp->val > taget->val)
tmp = tmp->left;
}
path.push_back(tmp->val);
return path;
}
};*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
TreeNode* ans=root;
while(1){
if(p->val<ans->val&&q->val<ans->val)
ans = ans->left;
else if(p->val>ans->val&&q->val>ans->val)
ans = ans ->right;
else
break;
}
return ans;
}
};
int main()
{
TreeNode node1(5);TreeNode node2(4);TreeNode node3(8);
TreeNode node4(2);TreeNode node5(6);TreeNode node6(9);
TreeNode node7(1);TreeNode node8(3);TreeNode node9(8);TreeNode node10(11);
node1.left=&node2;node1.right=&node3;
node2.left=&node4;node3.left=&node5;node3.right=&node6;
node4.left=&node7;node4.right=&node8;node6.left=&node9;node6.right=&node10;
Solution s;
TreeNode* ans = s.lowestCommonAncestor(&node1, &node2, &node8);
cout<<ans->val<<endl;
return 0;
}
今天也是爱zz的一天哦!