- 二叉搜索树与双向链表
- 题目
- 解题思路
二叉搜索树与双向链表
题目
牛客网
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
解题思路
- 由于 BST 的特性,采用中序遍历正好符合排序
- 要考虑 root 节点要与 左节点的最大值连接,与右节点的最小值连接
- 增加一个已排序链表的指针,指向最后一个已排序节点
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);
}