[进军硅谷]将二叉搜索树转化为一个有序的循环链表

来源:互联网 发布:豪世华邦网络办公平台 编辑:程序博客网 时间:2024/04/24 21:20

输入:一颗二叉搜索树
输出:有序的循环链表
思路:采用中序遍历,对于每一个遍历的节点,将该节点的左指针指向前面一个节点,与此同时,将前一个节点的右指针指向该遍历的节点。同时注意两点:

  • 没有对head节点进行赋值
  • 最后一个节点的右指针没有指向第一个节点

解决方法:每次递归调用的时候,更新当前节点的右指针指向head节点,更新head节点的左指针为当前节点。如果pre节点为空,则更新head节点为当前节点。
主要代码如下:

class TreeNode{    int val;    TreeNode left;    TreeNode right;    public TreeNode(int val){        this.val=val;        left=null;        right=null;    }}public class TreeToDoubleList{    public TreeNode treeToDouleList(TreeNode root){        TreeNode pre=null;        TreeNode head=null;        treeToDoubleList(root,pre,head);        return head;    }    //p为当前遍历节点    public void treeToDoubleList(TreeNode p,TreeNode pre,TreeNode head){        if(p==null) return;        treeToDoubleList(p.left,pre,head);        p.left=pre;//当前节点的左指针指向前一个遍历的节点;        if(pre!=null)//如果前一个节点不为空,则前一个节点的右指针指向该遍历的节点            pre.right=p;        else//如果为空,表示pre未被初始化,则头节点指向该遍历节点。            head=p;        TreeNode right=p.right;        //将首尾相连接,建立循环        head.left=p;        p.right=head;        //进入下一个节点之前,更新pre的值        pre=p;        treeToDoubleList(right,pre,head);           }}
阅读全文
0 0
原创粉丝点击