指 Offer 28. 对称的二叉树

116 阅读1分钟

题目:
请实现一个函数,用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样,那么它是对称的。

例如,二叉树 [1,2,2,3,4,4,3] 是对称的。

    1    /
2   2  / \ /
3  4 4  3 但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:

    1    /
2   2    \  
3    3

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/du… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:
方法一 dfs

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func isSymmetric(root *TreeNode) bool {
    if root == nil {
        return true 
    }
    return dfs(root.Left, root.Right)
}

func dfs(left, right *TreeNode) bool {
    if left == nil && right == nil {
        return true
    }
    if left == nil || right == nil || left.Val != right.Val {
        return false
    }

    return dfs(left.Left, right.Right) && dfs(left.Right, right.Left)
}

方法二 bfs

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func isSymmetric(root *TreeNode) bool {
    list := make([]*TreeNode, 0)
    list = append(list, root)
    return bfs(list)
}


func bfs(arr []*TreeNode) bool {

   for len(arr) != 0 {
        length := len(arr)
        if !symmetry(arr) {
            return false
        } 
        for i := 0; i < length; i ++ {
            if arr[i] == nil {
                continue
            }
            arr = append(arr, arr[i].Left) 
            arr = append(arr, arr[i].Right) 
        }
        // 左节点右节点都加入队列
        // 删除上一层的节点
        arr = arr[length:]       
   }
    
   return true


}


func symmetry(list []*TreeNode) bool {
    if len(list) < 2 {
        return true
    }
    i , j := 0, len(list) - 1
    for  i < j {
        if (list[i] == nil &&  list[j] == nil) || ( list[i] != nil &&  list[j] != nil && list[i].Val == list[j].Val) {

            i ++
            j --
        } else {

            return false
        }
    }
    return true
}