226. 翻转二叉树
给你一棵二叉树的根节点 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
层序遍历然后挨个节点做翻转就好了不难。
递归更简洁
func invertTree(root *TreeNode) *TreeNode {
if root==nil{
return nil
}
nodeSlice := []*TreeNode{root}
for len(nodeSlice)>0{
flag := len(nodeSlice)
for _, v := range nodeSlice{
v.Left, v.Right = v.Right, v.Left
if v.Left != nil{
nodeSlice = append(nodeSlice, v.Left)
}
if v.Right != nil{
nodeSlice = append(nodeSlice, v.Right)
}
}
nodeSlice = nodeSlice[flag:]
}
return root
}
时间O(n)。
空间是Slice的cap或者说是各层结点数的最大值。
func invertTree(root *TreeNode) *TreeNode {
if root == nil {
return nil
}
left := invertTree(root.Left)
right := invertTree(root.Right)
root.Left = right
root.Right = left
return root
}
作者:力扣官方题解
链接:https://leetcode.cn/problems/invert-binary-tree/solutions/415160/fan-zhuan-er-cha-shu-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
时间:O(n)
空间:最坏O(n)平均O(logn)