二叉树展开为链表的三种写法

二叉树展开为链表

二叉树展开为链表的三种写法_第1张图片

链表头节点法
新建一个树形链表,前序遍历这个树,遍历到一个节点就往里插

class Solution {
    TreeNode dummyNode = new TreeNode(0,null,null);
    TreeNode p = dummyNode;
    public void flatten(TreeNode root) {
        if(root == null) return;
        //线序遍历这个树,然后往里插就完事了
        TreeNode l = root.left;
        TreeNode r = root.right;
        p.right = root;
        p.right.left = null;
        p=p.right;
        flatten(l);
        flatten(r);
        //再指回去
        root = dummyNode.right;
    }
}

连接法
每次找左子树最右的节点,然后连接即可

class Solution {
    //每一次找左子树最右的节点,然后连接
    public void flatten(TreeNode root) {
        TreeNode cur = root;
        while(cur!=null){
            if(cur.left!=null){
                TreeNode leftMaxRight = cur.left;
                while(leftMaxRight.right!=null)
                    leftMaxRight = leftMaxRight.right;
                leftMaxRight.right = cur.right;
                cur.right = cur.left;
                cur.left =null;    
            }
            cur = cur.right;
        }
    }
}

倒着前序遍历法
这里有一种非常好的思路,就是在类里面定义一个变量,这个变量作为全局的变量,所有递归都能访问到,这样子就可以找到这个递归前一个节点了。

class Solution {
    TreeNode preNode;
    public void flatten(TreeNode root) {
        if(root==null) return ;
        flatten(root.right);
        flatten(root.left);
        root.left = null;
        root.right = preNode;
        preNode = root;
    }
}

你可能感兴趣的:(算法,链表,数据结构)