【算法】二叉搜索树与双向链表

题目

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

二叉树类

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }
}

分析

二叉搜索树的定义：
若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树。

思路

对于一颗二叉搜索树，由于他的左子树<根<右子树，由这个性质我们可以用中序遍历得到一个排序好的序列。然后再将该序列改造成一个双向链表即可

public class Convert {
    public TreeNode solution(TreeNode pRootOfTree){
        //判空
        if (pRootOfTree==null)  return null;

        //因为要频繁的插入，所以用链表存储中序遍历后的节点
        List<TreeNode> list = new LinkedList<>();
        //用于存放访问过的节点
        Stack<TreeNode> stack = new Stack<>();
        TreeNode node = pRootOfTree;
        //中序遍历
        while (node!=null || !stack.isEmpty()){
            while (node!=null){
                stack.push(node);
                node=node.left;
            }
            if (!stack.isEmpty()){
                node=stack.pop();
                list.add(node);
                node=node.right;
            }
        }
        //改造成双向链表，left指向前一个节点，right指向下一个节点
        for (int i=0; i<list.size(); i++){
            list.get(i).left = i==0? null : list.get(i-1);
            list.get(i).right = i==list.size()-1? null : list.get(i+1);
        }
        return list.get(0);

    }
}