求二叉樹的鏡像
遞歸
破壞原來的樹,把原來的樹改爲其鏡像
- 如果二叉樹爲空,則返回空;
- 如果不爲空,則求左子樹和右子樹的鏡像,然後交換左右子樹。
/**
* 求二叉樹的鏡像
* @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;
}
不能破壞原來的樹,返回一個新的鏡像
- 如果二叉樹爲空,則返回空;
- 如果不爲空,則創建一個新的節點,新節點的左右孩子分別等於原右子樹和左子樹的鏡像(這裏相當於交換了左右子樹)。
/**
* 求二叉樹的鏡像,不破壞原樹
* @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);
}
}
}