剑指 offer 树算法题:二叉搜索树与双向链表

题目描述:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表要求不能创建任何新的节点,只能调整树中节点指针的指向

分析:首先得了解二叉搜索树的特点,即左孩子 < 父节点 < 右孩子,二叉搜索树的中序遍历即为有序。要求只能在原树中改变指针指向。可以采用将原树中节点的右指针指向下一个节点,左指针指向上一个节点。

        中序遍历递归法

        1. 首先递归遍历左子树;

        2. 遍历到当前节点时,判断若是最左节点,说明是双向链表的头结点,否则将前驱节点的后驱指向当前遍历节点,当前节点的前驱指向 pre,更新前驱节点 pre 为 当前遍历节点。

         3. 然后递归遍历右子树;

         4. 最后递归结束,将头结点 head 的前驱指向最后一个节点 pre,将最后一个节点 pre 的后驱指向头结点 head。

        中序遍历迭代法,利用栈来迭代中序遍历,出栈时执行和递归中遍历当前节点相同的操作,同样,遍历结束后,将头结点 head 的前驱指向最后一个节点 pre,将最后一个节点 pre 的后驱指向头结点 head。

        除了遍历的同时改变指针指向,也可以先遍历放入数组,然后依次从前往后进行节点间双向指向,显然后者更易懂。

求解:

// 中序遍历递归法
function treeToDoublyList1(root: TreeNode<number> | null): TreeNode<number> | null {
  // 空树 返回 null
  if (root === null) return null;
  // 执行双向链表的头结点
  let head: any = null;
  // 当前遍历节点的前驱节点
  let pre: any = null;
  function dfs(node: TreeNode<number> | null): void {
    if (node === null) return;
    const { left, right } = node;
    // 遍历左子树
    dfs(left);
    // 处理当前节点
    if (pre === null) {
      // 最左节点 (或者用 head === null 判断)
      head = node;
    } else {
      // 前驱节点的后驱指向当前遍历节点
      pre.right = node;
    }
    // 当前节点的前驱节点为 pre
    node.left = pre;
    // 更新当前遍历节点的前驱节点
    pre = node;
    // 遍历右子树
    dfs(right);
  }
  dfs(root);
  // 设置头结点的前驱指向最后一个节点
  if (head) {
    head.left = pre;
  }
  // 设置最后一个节点的后驱指向头结点
  if (pre) {
    pre.right = head;
  }
  return head;
}

// 中序遍历迭代法
function treeToDoublyList2(root: TreeNode<number>  | null): TreeNode<number> | null {
  // 空树 返回 null
  if (root === null) return null;
  // 执行双向链表的头结点
  let head: any = null;
  // 当前遍历节点的前驱节点
  let pre: any = null;
  const stack: (TreeNode<number> | null)[] = [];
  let curNode: TreeNode<number> | null = root;
  while (stack.length || curNode !== null) {
    while (curNode !== null) {
      // 一直遍历到最左孩子节点
      stack.push(curNode);
      curNode = curNode.left;
    }
    if (stack.length !== 0) {
      // 栈非空
      let node = stack.pop()!;
      if (head === null) {
        // 最左节点
        head = node;
      } else {
        // 前驱节点的后驱指向当前遍历节点
        pre.right = node;
      }
      // 当前节点的前驱节点为 pre
      node.left = pre;
      // 更新当前遍历节点的前驱节点
      pre = node;
      curNode = node.right; // 继续遍历当前节点的右子树
    }
  }
  // 设置头结点的前驱指向最后一个节点
  if (head) {
    head.left = pre;
  }
  // 设置最后一个节点的后驱指向头结点
  if (pre) {
    pre.right = head;
  }
  return head;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

薛定谔的猫96

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值