剑指 Offer 36. 二叉搜索树与双向链表 思路与代码

 

 

思路：遇到类似的题，没见过没关系，不要慌！先分析它给出的结构与需要的结构各有什么特点？一定是根据各自的特点去写代码。可以看到给的数是一个二叉搜索树，二叉搜索树有什么特点呢？二叉搜索树的左子树都小于这个节点，右子树都大于这个节点。它的中序遍历是从小到大遍历的。题目需要的是什么结构？是一个双向的循环链表，而且是一个有顺序的。这个特点跟二叉搜索树中序遍历不谋而合，分析到这就很明朗了，十有八九就是利用中序遍历解决问题！！

中序线索化二叉树与这个类似。可以定义一个前置节点pre，使用flag记录第一个输出的节点head（最终返回它）。

在中序遍历每一个节点输出的时候，让pre.right指向此节点，this.left指向pre。（需要注意pre为空的情况。）最终循环结束的时候将head.left指向pre，pre.right指向head 返回head即可。

/*
// Definition for a Node.
class Node {
    public int val;
    public Node left;
    public Node right;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val,Node _left,Node _right) {
        val = _val;
        left = _left;
        right = _right;
    }
};
*/
class Solution {

    public Node treeToDoublyList(Node root) {
        if(root==null){
            return null;
        }
        Stack<Node> s1 = new Stack<>();
        //中序遍历修改
        boolean flag=false;
        Node head =null; 
        Node pre = null;
        while(root!=null || !s1.empty()){
        while(root!=null){
            s1.push(root);
            root=root.left;
        }
        root=s1.pop();
        if(flag==false){
            head=root;
            flag=true;
        }
        root.left=pre;
        if(pre!=null){
            pre.right=root;
        }
        pre=root;
        
        root=root.right;}

        //最后连上
        head.left=pre;
        pre.right=head;
        return head;
    }
   
}