树中两个节点的最低公共祖先
解决方法一:
假设是二叉搜索树(二叉搜索树是一个排序的二叉树,左子树的结点小于根结点,右子树的结点大于根结点),
故找到一个结点,使其大于左子结点小于右子结点即可。
public static TreeNode getLastCommonNode(TreeNode root, TreeNode p, TreeNode q) {
if (root == null || p == root || q == root)
return root;
if (p == null || q == null)
return null;
if (root.val > p.val && root.val > q.val)
return getLastCommonNode(root.left, p, q);
if (root.val < p.val && root.val < q.val)
return getLastCommonNode(root.right, p, q);
return root;
}
解决方法二:如果是普通二叉树,但有parent节点,
则可转化为两个指向root的单链表求第一个公共节点
1.用栈
2.先找出2个链表的长度,然后让长的先走两个链表的长度差,然后再一起走,直到找到第一个公共结点。
代码略
解决方法三:是普通二叉树
递归遍历找到所给定的两个结点,然后向上标记,直到有一个结点的左子结点和右子结点都不为空返回即可
public static TreeNode getLastCommonNode3(TreeNode root, TreeNode p, TreeNode q) {
if (root == null || p == root || q == root)
return root;
if (p == null || q == null)
return null;
TreeNode left = getLastCommonNode3(root.left, p, q);
TreeNode right = getLastCommonNode3(root.right, p, q);
if (left == null)
return right;
if (right == null)
return left;
return root;
}
解法四:迭代解法:
需要我们保存下由root根节点到p和q节点的路径,并且将路径存入list中,则问题转化为求两个list集合的最后一个共同元素。
public static TreeNode getLastCommonNode4(TreeNode pRoot, TreeNode p, TreeNode q) {
if (pRoot == null || p == pRoot || q == pRoot)
return pRoot;
if (p == null || q == null)
return null;
List<TreeNode> pathP = new ArrayList<>();
List<TreeNode> pathQ = new ArrayList<>();
pathP.add(pRoot);
pathQ.add(pRoot);
getPath(pRoot, p, pathP);
getPath(pRoot, q, pathQ);
TreeNode res = null;
for (int i = 0; i < pathP.size() && i < pathQ.size(); i++) {
if(pathP.get(i) == pathQ.get(i))
res = pathP.get(i);
else
break;
}
return res;
}
private static boolean getPath(TreeNode pRoot, TreeNode n, List<TreeNode> path) {
if(pRoot == n)
return true;
if(pRoot.left != null){
path.add(pRoot.left);
if(getPath(pRoot.left, n, path))
return true;
path.remove(path.size()-1);
}
if(pRoot.right != null){
path.add(pRoot.right);
if(getPath(pRoot.right, n, path))
return true;
path.remove(path.size()-1);
}
return false;
}