【剑指Offer】二叉搜索树与双向链表（二叉搜索树转换为双向链表，本地实现）

二叉搜索树与双向链表

题目描述

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

后序遍历+迭代+修改节点指针

方法一：非递归版
解题思路：
1. 核心是中序遍历的非递归算法。
2. 修改当前遍历节点与前一遍历节点的指针指向。
3. 注意找双向链表头

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.Stack;

public class Solution {
    
    //因为二叉搜索树，所以基于中序遍历
    public TreeNode Convert(TreeNode pRootOfTree) {
        if (pRootOfTree == null) {
            return null;
        }
        Stack<TreeNode> stack = new Stack<>();
        TreeNode p = pRootOfTree;
        TreeNode pre = null;// 用于保存中序遍历序列的上一节点
        boolean isFirst = true;
        while (p != null || !stack.isEmpty()) {
            //中序遍历，找到最左下节点(经过的路径上所有节点入栈)
            while (p != null) {
                stack.push(p);
                p = p.left;
            }
            p = stack.pop();
            //找转换成双向链表后的链表头
            if (isFirst) {
                // 将中序遍历序列中的第一个节点记为 pRootOfTree
                pRootOfTree = p;
                //链表头是下一个结点的前一个节点
                pre = pRootOfTree;
                isFirst = false;
            }else {
                pre.right = p;
                p.left = pre;
                pre = p;
            }
            p = p.right;
        }
        return pRootOfTree;
    }
}

后序遍历+递归+修改指针

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/

//直接用中序遍历
public class Solution {
    
    private TreeNode head = null;
    private TreeNode realHead = null;
    
    public TreeNode Convert(TreeNode pRootOfTree) {
        ConvertSub(pRootOfTree);
        return realHead;
    }
     
    private void ConvertSub(TreeNode pRootOfTree) {
        if(pRootOfTree == null) {
            return;
        }
        ConvertSub(pRootOfTree.left);
        if (head == null) {
            head = pRootOfTree;
            realHead = pRootOfTree;
        } else {
            head.right = pRootOfTree;
            pRootOfTree.left = head;
            head = pRootOfTree;
        }
        ConvertSub(pRootOfTree.right);
    }
}

参考：https://www.nowcoder.com/questionTerminal/947f6eb80d944a84850b0538bf0ec3a5?f=discussion