【数据结构与算法】leetcode236. 二叉树的最近公共祖先

leetcode236. 二叉树的最近公共祖先

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

百度百科中最近公共祖先的定义为：“对于有根树 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

---

解题：

• 利用一个hashmap1存储每个子节点的父节点
• 对p遍历父节点，并存在一个hashmap2中
• 对q遍历，若在hashmap2中能找到就返回

#include<vector>
#include<iostream>
#include<queue>
#include<unordered_map>
//#include<algorithm>
using namespace std;

struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int val):val(val),left(nullptr),right(nullptr){}
};
unordered_map<int, TreeNode*> map1;
unordered_map<int, bool> map2;
void DFS(TreeNode* root) {
	if (root == nullptr)return;
	if (root->left) {
		map1[root->left->val] = root;
		DFS(root->left);
	}
	if(root->right) {
		map1[root->right->val] = root;
		DFS(root->right);
	}
}

TreeNode* lowestCommon(TreeNode* root, TreeNode* p, TreeNode* q)
{
	map1[root->val] = nullptr;
	DFS(root);
	while (p != nullptr) {
		map2[p->val] = true;
		p = map1[p->val];
	}
	while (q != nullptr) {
		if (map2[q->val])return q;
		q = map1[q->val];
	}
}
int main() {
	TreeNode* root = new TreeNode(5);
	root->left = new TreeNode(9);
	root->right = new TreeNode(10);
	root->left->left = new TreeNode(8);
	root->left->right = new TreeNode(7);
	root->right->left = new TreeNode(2);
	root->right->right = new TreeNode(6);
	TreeNode* res = lowestCommon(root, root->left->left, root->right->right);
	cout << res->val << endl;
}