【力扣-数据结构】【第 12 天】226. 翻转二叉树

标题 226. 翻转二叉树
难度 简单
天数 第12天,第1/2题
数据结构

给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。

示例 1:

【力扣-数据结构】【第 12 天】226. 翻转二叉树_第1张图片

输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]

示例 2:

【力扣-数据结构】【第 12 天】226. 翻转二叉树_第2张图片

输入:root = [2,1,3]
输出:[2,3,1]

示例 3:

输入:root = []
输出:[]

提示:

  • 树中节点数目范围在 [0, 100]
  • -100 <= Node.val <= 100

以上内容来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

  1. 我们可以利用队列存储每一层二叉树
	Queue<TreeNode> queue = new LinkedList();
	queue.offer(root);
  1. 从队列里取出二叉树,只要将左树和右树交换,就能得到一个翻转二叉树
    • 先用leftNode记录左树
    • 再将右树赋值给左树
    • 再将leftNode赋值给右树
    • 这样就完成了当前树的一次翻转
	TreeNode node = queue.poll();
	TreeNode leftNode = node.left;
	node.left = node.right;
	node.right = leftNode;
  1. 如果交换完毕
    • 左子树存在,左子树添加进队列
    • 右子树存在,右子树添加进队列
	if(node.left != null){
		queue.offer(node.left);
	}
	if(node.right != null){
		queue.offer(node.right);
	}

【力扣-数据结构】【第 12 天】226. 翻转二叉树_第3张图片

最后完整代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    //数据结构 第 12 天 1/2 树
    public TreeNode invertTree(TreeNode root) {
        if(root == null){
            return root;
        }
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        while(!queue.isEmpty()){
           TreeNode node = queue.poll();
           TreeNode leftNode = node.left;
           node.left = node.right;
           node.right = leftNode;
           if(node.left != null){
               queue.offer(node.left);
           }
            if(node.right != null){
               queue.offer(node.right);
           }
        }
        return root;
    }
}

你可能感兴趣的:(力扣-算法(简单)合集,LeetCode-数据结构入门,leetcode,数据结构,算法,226.,翻转二叉树,java)