标题 | 226. 翻转二叉树 |
难度 | 简单 |
天数 | 第12天,第1/2题 |
数据结构 | 树 |
给你一棵二叉树的根节点
root
,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
示例 2:
输入:root = [2,1,3]
输出:[2,3,1]
示例 3:
输入:root = []
输出:[]
提示:
- 树中节点数目范围在
[0, 100]
内 -100 <= Node.val <= 100
以上内容来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
- 我们可以利用队列存储每一层二叉树
Queue<TreeNode> queue = new LinkedList();
queue.offer(root);
- 从队列里取出二叉树,只要将左树和右树交换,就能得到一个翻转二叉树
- 先用leftNode记录左树
- 再将右树赋值给左树
- 再将leftNode赋值给右树
- 这样就完成了当前树的一次翻转
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);
}
最后完整代码:
/**
* 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;
}
}