二叉搜索树与双向链表

题目

牛客网

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

解题思路

  1. 由于 BST 的特性,采用中序遍历正好符合排序
  2. 要考虑 root 节点要与 左节点的最大值连接,与右节点的最小值连接
  3. 增加一个已排序链表的指针,指向最后一个已排序节点

    public TreeNode Convert(TreeNode pRootOfTree) {
    if (pRootOfTree == null) {
        return null;
    }
    
    TreeNode[] nodeList = {new TreeNode(-1)};
    
    ConvertToLink(pRootOfTree, nodeList);
    
    TreeNode cursor = pRootOfTree;
    while (cursor.left != null) {
        cursor = cursor.left;
    }
    
    cursor.right.left = null;
    return cursor.right;
    }
    
    private void ConvertToLink(TreeNode root, TreeNode[] nodeList) {
    if (root == null) {
        return;
    }
    
    ConvertToLink(root.left, nodeList);
    
    root.left = nodeList[0];
    nodeList[0].right = root;
    nodeList[0] = root;
    
    ConvertToLink(root.right, nodeList);
    }