leetcode 226.翻转二叉树 golang实现

描述
翻转一棵二叉树。

示例:

输入:

     4
   /   \
  2     7
 / \   / \
1   3 6   9
输出:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

思路
1. 递归
先递归自己的左右树 然后交换自己的左右树
2. 非递归
遍历每一个节点 并将自己的左右节点交换
实现
//递归
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func invertTree(root *TreeNode) *TreeNode {
    if root == nil{
        return root
    }

    root.Left = invertTree(root.Left)
    root.Right = invertTree(root.Right)

    left := root.Left
    root.Left = root.Right
    root.Right = left
    
    return root
}

//非递归
func invertTree(root *TreeNode) *TreeNode {
    stack := make([]*TreeNode, 0)

    curNode := root
    for curNode != nil || len(stack) > 0{
        for curNode != nil{
            stack = append(stack, curNode)
            curNode = curNode.Left
        }

        top := stack[len(stack) - 1]
        stack = stack[0: len(stack) - 1]
        curNode = top.Right        // 注意 先将pop出的节点的右节点赋值之后 再交换左右树
        
        
        left := top.Left
        top.Left = top.Right
        top.Right = left
    }

    return root
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值