剑指Offer——二叉搜素树与双向链表

题目描述

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

解析

非递归解法

二叉搜索树的特征是左子树中节点的值小于根节点的值，右子树的节点的值大于根节点的值。所以中序遍历的顺序就是排序的双向链表从左向右的顺序。所以非递归中序遍历搜索二叉树，遍历过程中，节点保存在栈里面，以一个pre记录前一个节点，当前遍历的节点左left指向pre，pre的right指向当前节点。如果了解二叉树非递归中序遍历，应该比较好理解这种解法。https://blog.csdn.net/weixin_42582008/article/details/100639783 这篇博客中记录了二叉树的先序，中序，后续非递归遍历方法。

import java.util.Stack;
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
            if(pRootOfTree == null){
                return null;
            }
            Stack<TreeNode> stack = new Stack<>();
            TreeNode p = pRootOfTree;
            TreeNode pre = p;
            TreeNode res = p;
            boolean isFirst = true;
            while(p != null || !stack.empty()){
                while(p != null){
                    stack.push(p);
                    p=p.left;
                }
                p = stack.pop();
                if(isFirst == true){
                    //用res来表示左子树最左边的节点，也就是双向链表的头节点
                    res = p;
                    pre = p;
                    isFirst = false;
                }else{
                    p.left = pre;
                    pre.right = p;
                    pre = p;
                }
                p = p.right;
            }
        return res;
    }
}

递归解法

当然，递归遍历也可以解决这个问题，先递归地将左子树处理为排序的双向链表，返回双向链表的头节点，将根节点的left指向双向链表最右边的节点，这个节点的right指向根节点，再递归的处理右子树，将其处理为双向链表，将根节点的right指向双向链表的头节点，将双向链表头节left点指向根节点,之后返回整个双向链表的头节点。

public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        if(pRootOfTree == null){
            return null;
        }
        if(pRootOfTree.left == null && pRootOfTree.right == null){
            return pRootOfTree;
        }
        TreeNode left = Convert(pRootOfTree.left);
        //第一次把while错写成了if，检查了好久每检查出哪里错了 
        while(left != null && left.right != null){
            left = left.right;
        }
        if(left != null){
            pRootOfTree.left = left;
            left.right = pRootOfTree;
        }
        TreeNode right = Convert(pRootOfTree.right);
        if(right != null){
            pRootOfTree.right = right;
            right.left = pRootOfTree;
        }
        while(pRootOfTree != null && pRootOfTree.left != null){
            pRootOfTree = pRootOfTree.left;
        }
        return pRootOfTree;
         
    }
}