剑指offer之——二叉树的下一个结点
思路1:
①:通过给定的结点找出该树的根结点。
②:中序遍历该树并存储在数组中。
③:最后依次读取数组中的结点,找出二叉树的下一个结点。
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
public class Solution {
//集合用于存储结点
ArrayList<TreeLinkNode> list = new ArrayList<>();
public TreeLinkNode GetNext(TreeLinkNode pNode) {
//异常处理
if (pNode == null) {
return null;
}
//找到树的根节点
TreeLinkNode root = pNode;
while (root.next != null) {
root = root.next;
}
midOrder(root);
int len = list.size();
for (int i = 0; i < len; i ++) {
if (list.get(i) == pNode) {
return i == len - 1 ? null : list.get(i + 1);
}
}
return null;
}
//中序遍历
public void midOrder(TreeLinkNode root) {
if (root != null) {
midOrder(root.left);
list.add(root);
midOrder(root.right);
}
}
}
时间复杂度:找根结点,做最坏的打算为O(n),中序遍历为O(n),再次遍历数组最坏的时间复杂度为O(n),故总的时间复杂度为3O(n)。
空间复杂度:存储树中的所有结点,空间复杂度为O(n)。
思路2:
该节点存在右子树:则下一个结点为右子树的最左结点。
该节点右子树不存在:
①该结点为其父节点的左结点,下一个结点为其父结点。
②该节点为其父节点的右结点,寻找下一个结点比较麻烦,应该依次从该结点找其祖先结点,直到有一个结点是其父节点的左节点。
注:黄色为存在右子树的,红色为不存在右子树且结点为父结点的左结点,绿色为不存在右子树且结点为父结点的右结点。
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode) {
if (pNode == null) {
return null;
}
//存在右子树,下一个结点是右子树的最左结点
if (pNode.right != null) {
TreeLinkNode node = pNode.right;
while (node.left != null) {
node = node.left;
}
return node;
}
//不存在右子树
else {
//该结点是父节点的左子节点,下一个结点是其父节点
TreeLinkNode parent = pNode.next;
if (parent == null || parent.left == pNode) {
return parent;
}
//该节点是父节点的右子结点,下一个结点是,沿着其父节点一直往上找,直到找到一个祖先结点的左结点是其父节点
else {
TreeLinkNode pparent = parent.next;
while (pparent != null && pparent.left != parent) {
parent = pparent;
pparent = pparent.next;
}
return pparent;
}
}
}
}
时间复杂度:O(n)。
空间复杂度:O(1)。