Java二叉树转单链表

分析:

(1)要求转化为先序遍历顺序,这个很容易想到设置一个链表,然后先序遍历二叉树,把节点和指针记录在链表就好了

(2)题目要求in-place,也就是说不允许申请额外空间了,所以不能设置一个新的链表,只能在当前树(左右孩子指针)的结构上做手脚了,而且从图上看最后的二叉树都是只有右子树的。

由上图可知,如果右子树不为空,则右子树最后肯定为左子树最后一个靠右的孩子节点的右子树,而左子树最后成为整棵树的右子树。这样,首先判断左子树是否为空,不为空就寻找到树根的左孩子节点,然后寻找该节点是否有右孩子,如果有继续寻找,直到找到属于叶子节点的右孩子,此时,该节点的右子树“指向”当前树的右子树,并将当前左子树变为树根的右孩子,将整棵树左孩子置为空。最后,根节点“指向”根节点的右孩子,继续上述操作,直到整棵树遍历完即得到结果。

举个例子,比如只有2,3,4这三个节点,原先2左指针指向3,右指针指向4,现在不行了,需要把3指向4,断开2和4的连接,也就是要2的左孩子指向2的右孩子。然后形成一个递归的感觉。

 

public class S10 {

    public static void main(String[] args) {
        TreeNode treeNode1 = new TreeNode(1);
        TreeNode treeNode2 = new TreeNode(2);
        TreeNode treeNode3 = new TreeNode(3);
        TreeNode treeNode4 = new TreeNode(4);
        TreeNode treeNode5 = new TreeNode(5);
        TreeNode treeNode6 = new TreeNode(6);

        treeNode1.left = treeNode2;
        treeNode2.left = treeNode3;
        treeNode2.right = treeNode4;

        treeNode1.right = treeNode5;
        treeNode5.right = treeNode6;


        TreeOperation.show(treeNode1);

        flatten(treeNode1);
        TreeOperation.show(treeNode1);

    }


    public static void flatten(TreeNode root){
        if (root == null) return;
        if (root.left != null) {
            TreeNode cur = root.left;
            while (cur.right != null) {
                cur = cur.right;
            }
            cur.right = root.right;
            root.right = root.left;
            root.left = null;
        }
        flatten(root.right);
    }
}