正文

LeetCode27. 二叉樹的鏡像 請完成一個(gè)函數(shù)，輸入一個(gè)二叉樹，該函數(shù)輸出它的鏡像。

例如輸入：

4

/
2 7 / \ /
1 3 6 9 鏡像輸出：

4

/
7 2 / \ /
9 6 3 1

示例 1： 輸入：root = [4,2,7,1,3,6,9] 輸出：[4,7,2,9,6,3,1]

限制： 0 <= 節(jié)點(diǎn)個(gè)數(shù) <= 1000

解題思路

方法一：遞歸法

根據(jù)二叉樹鏡像的定義，考慮遞歸遍歷（dfs）二叉樹，交換每個(gè)節(jié)點(diǎn)的左 / 右子節(jié)點(diǎn)，即可生成二叉樹的鏡像。 遞歸解析： 終止條件： 當(dāng)節(jié)點(diǎn) rootroot 為空時(shí)（即越過葉節(jié)點(diǎn)），則返回 nullnull ； 遞推工作： 初始化節(jié)點(diǎn) tmptmp ，用于暫存 rootroot 的左子節(jié)點(diǎn)； 開啟遞歸 右子節(jié)點(diǎn) mirrorTree(root.right)mirrorTree(root.right) ，并將返回值作為 rootroot 的 左子節(jié)點(diǎn) 。 開啟遞歸 左子節(jié)點(diǎn) mirrorTree(tmp)mirrorTree(tmp) ，并將返回值作為 rootroot 的 右子節(jié)點(diǎn) 。 返回值： 返回當(dāng)前節(jié)點(diǎn) rootroot ；

Q： 為何需要暫存 rootroot 的左子節(jié)點(diǎn)？ A： 在遞歸右子節(jié)點(diǎn) “root.left = mirrorTree(root.right);root.left=mirrorTree(root.right);” 執(zhí)行完畢后， root.leftroot.left 的值已經(jīng)發(fā)生改變，此時(shí)遞歸左子節(jié)點(diǎn) mirrorTree(root.left)mirrorTree(root.left) 則會(huì)出問題。

class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;
        TreeNode tmp = root.left;
        root.left = mirrorTree(root.right);
        root.right = mirrorTree(tmp);
        return root;
    }
}

方法二：輔助棧（或隊(duì)列）

利用棧（或隊(duì)列）遍歷樹的所有節(jié)點(diǎn) nodenode ，并交換每個(gè) nodenode 的左 / 右子節(jié)點(diǎn)。 算法流程： 特例處理： 當(dāng) rootroot 為空時(shí)，直接返回 nullnull ； 初始化： 棧（或隊(duì)列），本文用棧，并加入根節(jié)點(diǎn) rootroot 。 循環(huán)交換： 當(dāng)棧 stackstack 為空時(shí)跳出； 出棧： 記為 nodenode ； 添加子節(jié)點(diǎn)： 將 nodenode 左和右子節(jié)點(diǎn)入棧； 交換： 交換 nodenode 的左 / 右子節(jié)點(diǎn)。 返回值： 返回根節(jié)點(diǎn) rootroot 。

class Solution {
    public TreeNode mirrorTree(TreeNode root) {
        if(root == null) return null;
        Stack<TreeNode> stack = new Stack<>() {{ add(root); }};
        while(!stack.isEmpty()) {
            TreeNode node = stack.pop();
            if(node.left != null) stack.add(node.left);
            if(node.right != null) stack.add(node.right);
            TreeNode tmp = node.left;
            node.left = node.right;
            node.right = tmp;
        }
        return root;
    }
}

以上就是java面試題解LeetCode27二叉樹的鏡像實(shí)例的詳細(xì)內(nèi)容，更多關(guān)于java面試LeetCode二叉樹鏡像的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論