剑指offer系列之二十五：二叉搜索树与双向链表

题目描述

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

由于二叉搜索树的中序遍历就是排序的，如果是构造单链表，只需要一次中序遍历就可以了，但现在需要构造双链表，也就是在中序遍历的过程中需要设置每个节点的left与right指针，现在问题是如何设置这两个指针？二叉搜索树有一个特点，就是根节点的左子树上所有节点都比根节点的值小，而右子树上的所有节点的值都比根节点的值大，利用这个性质，当遍历根节点的左孩子的时候，可以继续把其当做左子树的根节点，右孩子可以当做右子树的根节点，从而使用递归完成。

以左子树为例，依次访问节点的左孩子，当遍历到叶子节点的时候，递归结束，并把该叶子节点设为左子树转换的双链表的第一个节点，然后把其父节点链在其右边，设置left和right指针；如果父节点有右孩子，则继续对其右孩子继续转换，链在父节点的右边（父节点的右孩子肯定比父节点大）。这样当左右子树都转换完成后，返回双链表的第一个节点就可以了。

下面是基于这种思路的Java代码（已被牛客AC）：

package com.rhwayfun.offer;public class ConvertTreeToDLinkedList {    static class TreeNode {        int val = 0;        TreeNode left = null;        TreeNode right = null;        public TreeNode(int val) {            this.val = val;        }    }    public TreeNode Convert(TreeNode pRootOfTree) {        if(pRootOfTree == null) return null;        if(pRootOfTree.left == null && pRootOfTree.right == null) return pRootOfTree;        TreeNode pLastNodeOfLeftList = Convert(pRootOfTree.left);        TreeNode p = pLastNodeOfLeftList;        //定位到左子树双链表的最后一个节点        while(p != null && p.right != null){            p = p.right;        }        //将root节点追加到左子树双链表的最后一个节点        if(pLastNodeOfLeftList != null){            p.right = pRootOfTree;            pRootOfTree.left = p;        }        //转换右子树为双链表        TreeNode pLastNodeOfRightList = Convert(pRootOfTree.right);;        //将root追加至右子树双链表的最后一个节点        if(pLastNodeOfRightList != null){            pLastNodeOfRightList.left = pRootOfTree;            pRootOfTree.right = pLastNodeOfRightList;        }        //返回左子树的第一个节点        if(pLastNodeOfLeftList != null)            return pLastNodeOfLeftList;        return pRootOfTree;    }    public static void main(String[] args) {        TreeNode pRootOfTree = new TreeNode(10);        TreeNode node1 = new TreeNode(6);        TreeNode node2 = new TreeNode(14);        TreeNode node3 = new TreeNode(4);        TreeNode node4 = new TreeNode(8);        TreeNode node5 = new TreeNode(12);        TreeNode node6 = new TreeNode(16);        pRootOfTree.left = node1;        pRootOfTree.right = node2;        node1.left = node3;        node1.right = node4;        node2.left = node5;        node2.right = node6;        TreeNode node = new ConvertTreeToDLinkedList().Convert(pRootOfTree);        while(node != null){            System.out.print(node.val + " ");            node = node.right;        }    }}