剑指offer——树中两个节点的最低公共祖先（68题）

求树中两个节点的最低公共祖先，此题可由多道题构成。

变型一：

如果该树是二叉查找树，求树中A、B两个节点的最低公共祖先，方法如下：
1、如果遍历结点C的值比A、B结点的值都大，那么最低公共祖先在C的左子树上，于是到C的左子树上重复1、2、3
2、如果遍历结点C的值比A、B结点的值都小，那么最低公共祖先在C的左子树上，于是到C的右子树上重复1、2、3
3、否则遍历结点C即是结点A、B的最低公共祖先

变型二：

如果该树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两个节点的最低公共祖先，方法如下：
还是需要开辟辅助内存：
从根节点开始，利用dfs，寻找节点A，构建链1
从根节点开始，利用dfs，寻找节点B，构建链2
从链1和链2中寻找最低公共祖先。

#include<vector>
using namespace std;

struct TreeNode {
	int val;
	vector<TreeNode*> mChildren;
};

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;
}

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;
}

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);
}