面试题68：树中两个节点的最低公共祖先

该题有三种变型：

一、如果该树是二叉查找树，求树中A，B两个节点的最低公共祖先，方法如下：

1.从树的根节点和两个输入节点开始进行比较，如果当前节点C的值比A,B节点值都大，那么最低的共同节点一定在当前节点的左子树中

2.如果当前节点C的值比A,B节点值都小，那么最低的共同父节点一定在当前节点的右子树中

3.在树中从上到下找到的第一个在两个输入节点值之间的节点就是最低的公共祖先

 

二、如果该树M叉树，且每个节点均有指向父结点的指针，求树中A、B两个节点的最低公共祖先，方法如下：

转换成求两个链表的第一个公共节点，即

F->D->B->A

H->E->B->A

解题思想：

1、利用辅助内存，两个栈，FDBA先后入栈1，HEBA先后入栈2，两个栈同时将相同的栈顶元素出栈，从而找到第一个公共节点

2、两条链均设置一个遍历指针，从头到尾遍历，遇到尾节点又返回另一条链的头节点，再次遍历，最终一定相遇，代码如下：

struct ListNode {
	int val;
	ListNode* next;
	ListNode(int x):val(x),next(nullptr){}
};
ListNode* getCommonNode(ListNode* head1, ListNode* head2) {
	if (!head1 || !head2)
		return head1 ? head1 : head2;
	ListNode* cur1 = head1;
	ListNode* cur2 = head2;
	while (cur1||cur2) {
		if (!cur1)
			cur1 = head2;
		if (!cur2)
			cur2 = head1;
		if (cur1 == cur2)
			return cur1;
		cur1 = cur1->next;
		cur2 = cur2->next;
	}
	return cur1;
}

三、

如果该树M叉树，没有父结点的指针，求树中A、B两个节点的最低公共祖先，方法如下：

还是需要开辟辅助内存：

从根节点开始，利用前序遍历，寻找节点A，构建链1

从根节点开始，利用前序遍历，寻找节点B，构建链2

从链1和链2中寻找最低公共祖先。

 

为了得到从根节点开始到输入的两个节点的两条路径，需要遍历两次树，每遍历一次的时间复杂度是O(n)，得到的两条路径的长度在最差情况下是O(n)，通常两条路径的长度是O(logn)(空间复杂度)

 

#include<vector>
using namespace std;
 
struct TreeNode {
	int val;
	vector<TreeNode*> mChildren;
};

//用来得到从根节点root开始到达节点pNode的路径，这条路径保存在path数组中
bool getNodePath(TreeNode* root, TreeNode* node, vector<TreeNode*>& path) {
	if (root == node)
		return true;
	path.push_back(root);
	bool found = false;
	vector<TreeNode*>::iterator it = root->mChildren.begin();
	while (!found&&it != root->mChildren.end()) {//dfs
		found = getNodePath(*it, node, path);
		it++;
	}
	if (!found) {//如果没有找到，将此当前尾结点弹出
		path.pop_back();
	}
	return found;
}

//用来得到两条路径path1和path2的最后一个公共节点
TreeNode* getLastCommonNode(const vector<TreeNode*>& path1, const vector<TreeNode*>& path2) {
	int size = path1.size() < path2.size() ? path1.size() : path2.size();
	TreeNode* commonNode=nullptr;
	for (int i = 0; i < size; ++i) {
		if (path1[i] == path2[i])
			commonNode = path1[i];
	}
	return commonNode;
}

//首先调用GetNodePath得到path1和path2，接着调用getLastCommonNode得到
//path1和path2的最后一个公共节点，即我们要找的最低公共祖先
TreeNode* getLastCommonParent(TreeNode* root, TreeNode* node1, TreeNode* node2) {
	if (!root || !node1 || !node2)
		return nullptr;
	vector<TreeNode*> path1;
	vector<TreeNode*> path2;
	getNodePath(root, node1, path1);
	getNodePath(root, node2, path2);
	return getLastCommonNode(path1, path2);
}

参考博客：

https://blog.csdn.net/TT_love9527/article/details/82116345