给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例 1:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。
示例 2:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。
示例 3:
输入:root = [1,2], p = 1, q = 2
输出:1
提示:
树中节点数目在范围 [2, 105] 内。
-109 <= Node.val <= 109
所有 Node.val 互不相同 。
p != q
p 和 q 均存在于给定的二叉树中。
思路:
定义两个数list p_vec 与 q_vec ,分别 表示从 root 到 p 与 q 的路径,用回溯法求路径
1. 分别从 下标 0 开始遍历到 下标 i ,若是 p_vec[ i+1] != q_vec[ i+1 ] ,说明 p_vec[ i ] == q_vec[ i ] , 即 p_vec[ i ] 或者 q_vec[ i ] 为题目所求,二叉树的最近公共祖先
c++
/**
* 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 == nullptr) {
return nullptr;
}
vector<TreeNode*> p_vec;
vector<TreeNode*> q_vec;
vector<TreeNode*> tmp_p;
vector<TreeNode*> tmp_q;
// 3 5 6
// 3 5 6
findPath(root, p, tmp_p, p_vec);
findPath(root, q, tmp_q, q_vec);
int len = p_vec.size() > q_vec.size() ? q_vec.size():p_vec.size();
int i = 0;
for(; i<len-1; i++) {
if(p_vec[i+1] != q_vec[i+1]) {
break;
}
}
cout << p_vec[0]->val << endl;
cout << p_vec.size() <<" " << i << endl;
return p_vec[i];
}
void findPath (TreeNode* root, TreeNode* target, vector<TreeNode*>& tree_vec, vector<TreeNode*>& result_vec) {
if(root == nullptr) {
return;
}
tree_vec.push_back(root);
if(root == target) {
vector<TreeNode*> tmp = tree_vec;
result_vec = tmp;
}
findPath(root->left, target, tree_vec, result_vec);
findPath(root->right, target, tree_vec, result_vec);
tree_vec.pop_back();
}
};