题目描述:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
分析:首先得了解二叉搜索树的特点,即左孩子 < 父节点 < 右孩子,二叉搜索树的中序遍历即为有序。要求只能在原树中改变指针指向。可以采用将原树中节点的右指针指向下一个节点,左指针指向上一个节点。
中序遍历递归法,
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;
}