二叉树的下一个结点(Java)

二叉树的下一个结点

http://www.nowcoder.com/questionTerminal/9023a0c988684a53960365b889ceaf5e

向定位根节点,然后使用中序遍历的方法将节点存入一个List中。从list中找到匹配的节点,输出List的下一个元素即可。

import java.util.ArrayList;
public class Solution {
    ArrayList<TreeLinkNode> list = new ArrayList<>();
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        TreeLinkNode temp = pNode;
        // 先定位根节点
        while(temp.next != null) {
            temp = temp.next;
        }
        addNode(temp);

        for(int i=0; i<list.size(); i++) {
            if(list.get(i) == pNode) {
                if(i != list.size()-1) {
                    return list.get(i+1);
                }
                return null;
            }
        }
        return null;
    }

    public void addNode(TreeLinkNode pNode) {
        if(pNode != null) {
            addNode(pNode.left);
            list.add(pNode);
            addNode(pNode.right);
        }
    }
}
全部评论

相关推荐

点赞 评论 收藏
分享
真烦好烦真烦:豆包润色了自己没看看吗,再说了,都说豆包是愚蠢且勤快的大学生,ds才是聪明的研究生,怎么敢让豆包写论文的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务