题解 | #二叉搜索树与双向链表#TOP30

思路:
1.利用中序遍历,TOP24 题解。因为二叉搜索树,左边的节点比根节点小,右边的节点比根节点大。所以排序的话不就是 左节点 根节点 右节点嘛,这不就是中序遍历树嘛
2.只需要得到了当前节点,我们保存前一个节点,每次将前一个节点的右节点指向当前节点,当前节点的左节点指向前一个节点即可,这就是双向链表了

import java.util.*;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        if (pRootOfTree == null) {
            return null;
        }
        //中序遍历  左 根 右
        Stack<TreeNode> stack = new Stack<>();
        TreeNode currentNode = pRootOfTree;
        TreeNode head = null, temp = null;
        while (!stack.isEmpty() || currentNode != null) {
            //寻找到所有左分支上的所有节点,入栈
            while (currentNode != null) {
                stack.push(currentNode);
                currentNode = currentNode.left;
            }
            //左分支上的最后一个节点
            currentNode = stack.pop();
            //如果temp是空的,说明第一次遍历 设置一下头节点
            if (temp == null) {
                temp = currentNode;
                head = temp;
            } else {
                //否则的话 当前节点的左节点,指向上一个节点,上一个节点的右节点指向当前节点
                currentNode.left = temp;
                temp.right = currentNode;

                temp = currentNode;
            }
            currentNode = currentNode.right;
        }
        return head;
    }
}
全部评论

相关推荐

美团 后端开发 总包n(15%是股票)
点赞 评论 收藏
分享
牛客5655:其他公司的面试(事)吗
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-26 15:46
已编辑
字节国际 电商后端 24k-35k
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务