[剑指Offer]二叉树的镜像(Java)

剑指Offer题目

二叉树的镜像 -- newcoder 剑指Offer 18

题目描述

操作给定的二叉树，将其变换为源二叉树的镜像。

输入描述:

二叉树的镜像定义：
源二叉树 
    	    8
    	   /  \
    	  6   10
    	 / \  / \
    	5  7 9 11
镜像二叉树
    	    8
    	   /  \
    	  10   6
    	 / \  / \
    	11 9 7  5

思路

 * 思路1：
 * 1、交换root节点的左右子树
 * 2、递归交换root.left和root.right的左右子树 
 * 
 * 思路2： 
 * 1、非递归处理
 * 2、层次遍历，每次处理当前元素，交换当前元素的左右子树即可

代码

package com.codinginterviews.tree;

import java.util.LinkedList;
import java.util.Queue;

/**
 * 题目： 
 * 二叉树的镜像 -- newcoder 剑指Offer 18
 * 
 * 题目描述：
 * 
操作给定的二叉树，将其变换为源二叉树的镜像。

 * 输入描述:
二叉树的镜像定义：
源二叉树 
    	    8
    	   /  \
    	  6   10
    	 / \  / \
    	5  7 9 11
镜像二叉树
    	    8
    	   /  \
    	  10   6
    	 / \  / \
    	11 9 7  5
 */
public class Mirror {

	static class TreeNode {
		int val;
		TreeNode left;
		TreeNode right;
		TreeNode(int x) { val = x; }
	}

	/**
	 * 思路：
	 * 1、交换root节点的左右子树
	 * 2、递归交换root.left和root.right的左右子树 
	 */
    public void mirror(TreeNode root) {
        if (root == null || (root.left == null && root.right == null)) {
        	return;
        }
        
        // 交换左右子树
        TreeNode tmp = root.left;
        root.left = root.right;
        root.right = tmp;
        
        // 处理root.left及root.right
        mirror(root.left);
        mirror(root.right);
    }
	
    /**
     * 思路： 
     * 1、非递归处理
     * 2、层次遍历，每次处理当前元素，交换当前元素的左右子树即可
     */
    public void mirrorII(TreeNode root) {
        if (root == null || (root.left == null && root.right == null)) {
        	return;
        }
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        while (!queue.isEmpty()) {
        	TreeNode cur = queue.poll();
        	
        	if (cur.left != null || cur.right != null) {
                // 交换左右子树
                TreeNode tmp = cur.left;
                // 此处必须赋值给cur.left/right，更新节点指向，不能使用临时变量
                cur.left = cur.right;
                cur.right = tmp;
        	}
        	
        	if (cur.left != null) {
        		queue.offer(cur.left);
        	}
        	if (cur.right != null) {
        		queue.offer(cur.right);
        	}
        }
        
    }
    
	public static void main(String[] args) {
		TreeNode root = new TreeNode(1);
		TreeNode node1 = new TreeNode(2);
		TreeNode node2 = new TreeNode(3);
		root.left = node1;
		root.right = node2;
		
		new Mirror().mirror(root);
		
		System.out.println(root.left.val);
	}

}