剑指offer刷题-二叉搜索树与双向链表（树的中序遍历）

题目描述

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

思路：

二叉搜索树将其转换为双向链表如上图所示。

根据二叉搜索树的特点：左结点的值<根结点的值<右结点的值，我们不难发现，使用二叉树的中序遍历出来的数据的数序，就是排序的顺序。因此，首先，确定了二叉搜索树的遍历方法。

接下来，我们看下图，我们可以把树分成三个部分：值为10的结点、根结点为6的左子树、根结点为14的右子树。根据排序双向链表的定义，值为10的结点将和它的左子树的最大一个结点链接起来，同时它还将和右子树最小的结点链接起来。

按照中序遍历的顺序，当我们遍历到根结点时，它的左子树已经转换成一个排序的好的双向链表了，并且处在链表中最后一个的结点是当前值最大的结点。我们把值为8的结点和根结点链接起来，10就成了最后一个结点，接着我们就去遍历右子树，并把根结点和右子树中最小的结点链接起来。

所以问题关键是需要确定上一个结点与当前结点的

给定的形式，输入为根节点，返回的是转换好的双向链表头结点。

方法一：非递归版解题思路：

1.核心是中序遍历的非递归算法。
2.修改当前遍历节点与前一遍历节点的指针指向。

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
import java.util.Stack;
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        if(pRootOfTree == null)
            return pRootOfTree;
        TreeNode p = pRootOfTree;
        TreeNode pre = null;
        boolean isFirst = true;
        while(p != null || !stack.isEmpty())
        {
            while(p != null)
            {
                stack.push(p);
                p = p.left;
            }
            p = stack.pop();
            if(isFirst)
            {
                pRootOfTree = p;   
                pre = pRootOfTree;
                isFirst = false;
            }
            else
            {
                pre.right = p;
                p.left = pre;   //确定链表关系
                pre = p;
            }
            p = p.right;
        }
        return pRootOfTree;
    }
}

方法二：递归版解题思路：

1.将左子树构造成双链表，并返回链表头节点。
2.定位至左子树双链表最后一个节点。
3.如果左子树链表不为空的话，将当前root追加到左子树链表。
4.将右子树构造成双链表，并返回链表头节点。
5.如果右子树链表不为空的话，将该链表追加到root节点之后。
6.根据左子树链表是否为空确定返回的节点。

/**
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 pRootOfTree;
        if(pRootOfTree.left == null && pRootOfTree.right == null)
            return pRootOfTree;
        //1.将左子树构造成双链表，并返回链表头节点
        TreeNode left = Convert(pRootOfTree.left);
        TreeNode p = left;
        //2.定位至左子树双链表最后一个节点
        while(p != null && p.right != null)
        {
            p = p.right;
        }
        //3.如果左子树链表不为空的话，将当前root追加到左子树链表
        if(left != null)
        {
            p.right = pRootOfTree;
            pRootOfTree.left = p;
        }
        // 4.将右子树构造成双链表，并返回链表头节点
        TreeNode right = Convert(pRootOfTree.right);
        // 5.如果右子树链表不为空的话，将该链表追加到root节点之后
        if(right != null)
        {
            pRootOfTree.right = right;
            right.left = pRootOfTree;
        }
        //若左链表的头节点为空，则返回根节点，否则为左链表的头节点
        return left != null ? left:pRootOfTree;
    }
}