剑指 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;
}