使用 Go 解決 LeetCode 問題:101. Symmetric Tree

Description

Given a binary tree, check whether it is a mirror of itself (ie, symmetric around its center).

For example, this binary tree [1,2,2,3,4,4,3] is symmetric:

1
2
3
4
5
    1
/ \
2 2
/ \ / \
3 4 4 3

But the following [1,2,2,null,3,null,3] is not:

1
2
3
4
5
  1
/ \
2 2
\ \
3 3
  • Note:

Bonus points if you could solve it both recursively and iteratively.

Solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
// TreeNode struct
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}

func isSymmetric(root *TreeNode) bool {
// 判斷樹根是否為空
if root == nil {
return true
}

// 判斷左右節點是否對稱
return rec(root.Left, root.Right)
}

func rec(left *TreeNode, right *TreeNode) bool {
// 判斷兩樹是否同時為空
if left == nil && right == nil {
return true
}

// 判斷兩樹是否其一為空
if left == nil || right == nil {
return false
}

// 判斷兩樹其值是否相同
if left.Val != right.Val {
return false
}

// 判斷左右節點是否對稱
return rec(left.Left, right.Right) && rec(left.Right, right.Left)
}

Note

假設有以下參數:

1
2
p: [1, 2, 2]
q: [1, 2, 2]

說明:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
一開始,判斷 root 是否為空,再判斷左右節點是否對稱。

判斷 left 的左節點和 right 的右節點是否對稱,以及 left 的右節點和 right 的左節點是否對稱,等待返回。

------------------------------
1(p)
/ \
2(a) 2(b)
------------------------------

左節點開始遞迴:

判斷 left 的左節點和 right 的右節點是否對稱,由於同時為空,返回 true

左節點的遞迴返回 true

右節點開始遞迴:

判斷 left 的右節點和 right 的左節點是否對稱,由於同時為空,返回 true

右節點的遞迴返回 true

最終返回:true

Code