思路一:
因为每个结点都包含指向父节点的指针,所以可以先找到父节点,然后再进行中序遍历,找到给点结点的下一个结点。
import java.util.ArrayList;
/*
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)
{
//1.先找到根结点
TreeLinkNode root = findRoot(pNode);
//2.中序遍历找到下一个结点
ArrayList<TreeLinkNode> linkNodes = new ArrayList<TreeLinkNode>();
inOrder(root,linkNodes);
int index = 0;
for (int i = 0; i < linkNodes.size(); i++) {
if (linkNodes.get(i).val == pNode.val){
index = i + 1;
break;
}
}
return index >= linkNodes.size() ? null : linkNodes.get(index);
}
private void inOrder(TreeLinkNode root, ArrayList<TreeLinkNode> linkNodes) {
if (root != null){
inOrder(root.left,linkNodes);
linkNodes.add(root);
inOrder(root.right,linkNodes);
}
}
private TreeLinkNode findRoot(TreeLinkNode pNode) {
if (pNode.next == null){
return pNode;
}
return findRoot(pNode.next);
}
}
思路二:
若该节点存在右子树:则下一个节点为右子树最左子节点。
若该节点不存在右子树:这时分两种情况:
- 该节点为父节点的左子节点,则下一个节点为其父节点
- 该节点为父节点的右子节点,则沿着父节点向上遍历,直到找到一个节点的父节点的左子节点为该节点,则该节点的父节点为下一个节点
import java.util.ArrayList;
/*
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;
}
//1.若该节点存在右子树:则下一个节点为右子树最左子节点
if (pNode.right != null){
pNode = pNode.right;
while (pNode.left != null){
pNode = pNode.left;
}
return pNode;
}else if (pNode.next != null && pNode == pNode.next.left){
//2. 该节点为父节点的左子节点,则下一个节点为其父节点
return pNode.next;
}else if (pNode.next != null && pNode == pNode.next.right){
//3.该节点为父节点的右子节点,则沿着父节点向上遍历,知道找到一个节点的父节点的左子节点为该节点,则该节点的父节点下一个节点
while (pNode.next != null && pNode.next.left != pNode){
pNode = pNode.next;
}
return pNode.next;
}else {
return pNode.next;
}
}
}