牛客网剑指offer
二叉搜索树与双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
相关结构
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
解法思想
二叉搜索树:树中任一节点的左子节点的值永远小于该节点值,右子节点的值永远大于该节点的值
中序遍历:按照左子节点-根节点-右子节点的顺序遍历树
解法
public class Solution {
private TreeNode pre, head;
public TreeNode Convert(TreeNode pRootOfTree) {
if (pRootOfTree == null) {
return null;
}
check(pRootOfTree);
return head;
}
private void check(TreeNode treeNode) {
if (treeNode.left != null) {
check(treeNode.left);
}
if (pre == null) {
head = treeNode;
} else {
pre.right = treeNode;
}
treeNode.left = pre;
pre = treeNode;
if (treeNode.right != null) {
check(treeNode.right);
}
}
}