剑指offer-26.二叉搜索树与双向链表

题目描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

题目分析

二叉搜索树，中序遍历为递增的有序序列。
树的结点结构：lchild data rchild
双链表结点结构：pre data next
转换为双链表，原左孩子指针指向当前节点前驱，右孩子指向当前节点后继

二叉树中序遍历有两种方式，递归和非递归

非递归

import java.util.*;
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        //空树，返回空
        if(pRootOfTree==null)
            return null;
        
        TreeNode p = pRootOfTree; //当前节点
        TreeNode pre = null;//当前节点的前驱
        TreeNode head = null;
        
        boolean flag = true;
        
        //1.栈不空或者p不空，循环
        while(!stack.empty()||p!=null){
            
            //一路向左，入栈
            while(p!=null){
                stack.push(p); //入栈
                p = p.left;               
            }
            
            //当前节点出栈
            p = stack.pop();
            
            //第一次出栈点,即双链表第一个节点无前驱,返回头节点
            if(flag){
                head = p;
                pre = p;
                flag = false;
            }
            //除第一个点外
            else{
                pre.right = p; //pre后继是p
                p.left = pre;
                pre = p;
            }
            
            p = p.right;
        }
        return head;
        
    }
}

递归

//中序遍历
//二叉树，左孩子前驱，右孩子后继，类似于线索二叉树
//中序遍历递归/非递归
//递归写法
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        if(pRootOfTree==null)
            return null;
        
        //若为叶子结点，返回
        if(pRootOfTree.left==null && pRootOfTree.right==null)
            return pRootOfTree;
        
        //1.返回双链表的头节点——从根结点开始一路向左，树最左边的结点
        TreeNode left= Convert(pRootOfTree.left);
        TreeNode cur = left;//当前双链表第一个结点
        
        //2.左子树双链表最后一个结点,根节点左孩子一路向右
        while(cur!=null&&cur.right!=null){
            cur = cur.right;
        }
        
        //3.左子树链表不空，当前root放入左子树链表中
        if(left!=null){
            cur.right = pRootOfTree;
            pRootOfTree.left = cur;
        }
        
        //4.右子树构造为双链表并返回头节点               
        TreeNode right = Convert(pRootOfTree.right);
        TreeNode cur2 = right;
        
        //5.右子树链表不空，当前节点放入右子树链表中
        if(right!=null){
            cur2.left = pRootOfTree;
            pRootOfTree.right = cur2;
        }
        
        if(left!=null)
            return left;
        else
            return pRootOfTree;
    }
}