求二叉樹的鏡像(遞歸+非遞歸)

求二叉樹的鏡像


遞歸

破壞原來的樹,把原來的樹改爲其鏡像
  1. 如果二叉樹爲空,則返回空;
  2. 如果不爲空,則求左子樹和右子樹的鏡像,然後交換左右子樹。
    /**
     * 求二叉樹的鏡像
     * @param root
     * @return
     */
    public TreeNode mirror(TreeNode root){
        if (root == null){
            return null;
        }
        TreeNode l = mirror(root.left);
        TreeNode r = mirror(root.right);
        root.left = r;
        root.right = l;
        return root;
    }
不能破壞原來的樹,返回一個新的鏡像
  1. 如果二叉樹爲空,則返回空;
  2. 如果不爲空,則創建一個新的節點,新節點的左右孩子分別等於原右子樹和左子樹的鏡像(這裏相當於交換了左右子樹)。
    /**
     * 求二叉樹的鏡像,不破壞原樹
     * @param root
     * @return
     */
    public TreeNode mirrorCopy(TreeNode root){
        if (root == null){
            return null;
        }
        TreeNode newTree = new TreeNode(root.val);
        newTree.left = mirror(root.right);
        newTree.right = mirror(root.left);
        return newTree;
    }

非遞歸實現

非遞歸實現,其實就是遍歷二叉樹的所有節點,然後交換每個節點的左右子樹即可。

這裏我們用的是層序遍歷的方式:

    /**
     * 求二叉樹的鏡像,非遞歸
     * @param root
     */
    public void mirror2(TreeNode root){
        if (root == null){
            return;
        }
        LinkedList<TreeNode> list = new LinkedList();
        list.add(root);
        while (!list.isEmpty()){
            TreeNode node = list.poll();
            TreeNode tmp = node.left;
            node.left = node.right;
            node.right = tmp;
            if (node.left != null){
                list.add(node.left);
            }
            if (node.right != null){
                list.add(node.right);
            }
        }
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章