一、题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
二、代码分析
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
//直接用中序遍历
public class Solution {
TreeNode head = null;
TreeNode realHead = null;//双向链表的头结点
public TreeNode Convert(TreeNode pRootOfTree) {
//第一次运行时,它会使最左边叶子节点为链表第一个节点
Convert(pRootOfTree.left);
if (head == null) {
head = pRootOfTree;
realHead = pRootOfTree;
} else {
//把根节点插入到双向链表右边,rightHead向后移动
head.right = pRootOfTree;
pRootOfTree.left = head;
head = pRootOfTree;
}
//把右叶子节点也插入到双向链表(rightHead已确定,直接插入)
Convert(pRootOfTree.right);
//返回左边头结点
return realHead;
}
}
运行结果: