[2021校招必看之Java版《剑指offer》-26] 二叉搜索树与双向链表

1、题目描述

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

2、解题思路

2.1 中序遍历

  看到题目关键词“二叉搜索树”、“排序”就可以快速反应肯定要用到前序遍历。
  题目要求不能创建新的结点，因此我们可以考虑创建一个List来保存二叉搜索树中序遍历的结果，然后遍历修改List中的每一个结点即可。

2.2 线索化二叉树

  之前我的博客在介绍数的基础时，说过线索二叉树的概念，大家可以前往浏览https://blog.csdn.net/qq_29051413/article/details/105495944，线索二叉树一般是用于解决叶子指针的左右指针为空的问题，把叶子指针指向某种遍历下的下一个结点。
  参考线索化二叉树的概念，我们需要定义三个结点引用：cur、pre和head。三者分别表示中序遍历的当前结点、中序遍历的当前结点的上一个结点以及最终返回的双向链表的头结点。再定义一个flag表示是否确定了head。
  按中序遍历的顺序对每一个结点cur进行线索化，线索化的步骤为：
  1、先判断head是否已经确定。如果未确定，且当前结点的左节点为空，那么当前结点就是head；
  2、线索化左子树cur.left；
  3、处理当前结点cur：
  3.1 当前结点cur的左节点为前驱结点pre；
  3.2 如果前驱结点pre不为null，则前驱结点的下一个结点pre.right为当前结点cur；
  3.3 更新前驱结点pre为当前结点cur，然后处理中序遍历的下一个结点；
  4、线索化右子树。

3、解题代码

3.1 中序遍历

package pers.klb.jzoffer.medium;

import java.util.ArrayList;
import java.util.List;

/**
 * @program: JzOffer2021
 * @description: 二叉搜索树与双向链表
 * @author: Meumax
 * @create: 2020-07-09 10:16
 **/
public class BstToDll {

    public TreeNode Convert(TreeNode pRootOfTree) {

        if (pRootOfTree == null || (pRootOfTree.left == null && pRootOfTree.right == null)) {
            return pRootOfTree;
        }

        List<TreeNode> list = new ArrayList<>();

        // 中序遍历的结果就是结点从小到大排序
        preoder(pRootOfTree, list);

        // 修改每一个结点的左右指针
        for (int i = 0; i < list.size(); i++) {
            if (i == 0) {
                list.get(i).left = null;
                list.get(i).right = list.get(i + 1);
            } else if (i == list.size() - 1) {
                list.get(i).left = list.get(i - 1);
                list.get(i).right = null;
            } else {
                list.get(i).left = list.get(i - 1);
                list.get(i).right = list.get(i + 1);
            }
        }

        return list.get(0);
    }

    /**
     * 中序遍历
     * @param root  遍历到的结点
     * @param list  把遍历到的结点存储到 list 中
     */
    private void preoder(TreeNode root, List<TreeNode> list) {
        if (root == null) {
            return;
        }
        preoder(root.left, list);
        list.add(root);
        preoder(root.right, list);
    }
}

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

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

3.2 线索化二叉树

package pers.klb.jzoffer.medium;

/**
 * @program: JzOffer2021
 * @description: 二叉搜索树与双向链表
 * @author: Meumax
 * @create: 2020-07-09 10:16
 **/
public class BstToDll {

    private TreeNode pre = null;    // 前驱结点
    private TreeNode head = null;    // 双向链表的头结点
    private boolean flag = false;   // 是否找到了head

    public TreeNode Convert(TreeNode pRootOfTree) {
        head = pRootOfTree;
        threadNode(pRootOfTree);
        return head;
    }

    /**
     * 线索化以 node 为根结点的二叉树
     *
     * @param cur
     */
    private void threadNode(TreeNode cur) {
        if (cur == null) {
            return;
        }

        // 确定head
        if (cur.left == null && !flag) {
            head = cur;
            flag = true;    // 已经找到了head
        }

        // 线索化左子树
        threadNode(cur.left);

        cur.left = pre;    // 当前结点的左节点为前驱结点
        if (pre != null) pre.right = cur;   // 前驱结点的有节点为当前结点
        pre = cur; // 当前结点更新的下一个结点的前驱结点

        // 线索化右子树
        threadNode(cur.right);
    }
}

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

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

4、解题心得

  本题两个解决思路在性能上差别不大，第一个更为简单，在笔试的时候优先使用。当然，在面试的时候能说出线索化二叉树能给自己加分。