題目:輸入一棵二叉搜索樹,將該二叉搜索樹轉換成一個排序的雙向鏈表。要求不能創建任何新的結點,只能調整樹中結點指針的指向。
思路:
因為要構成排序的雙向鏈表且是二叉搜索樹,所以可以通過中序遍歷完成。遍歷時,將節點的左子樹指向前一個節點,右子樹指向后一個節點即可。
源碼:GitHub源碼
/**
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 pre = null;
public TreeNode Convert(TreeNode root) {
if(root == null) return root;
InOrder(root);
return Head;
}
public void InOrder(TreeNode root){
if(root == null) return;
InOrder(root.left);
if(pre == null){
pre = root;
Head = root;
}else{
pre.right = root;
root.left = pre;
pre = root;
}
InOrder(root.right);
}
}