白話解 Leetcode - 111 Minimum Depth of Binary Tree

Posted by Alan Zhan on Thursday, June 16, 2022

111. Minimum Depth of Binary Tree

https://leetcode.com/problems/minimum-depth-of-binary-tree/


題意

尋找這棵樹最小的節點,他的深度為何?

解題思路

這次的題目與 104 Maximum Depth of Binary Tree 相似,這次當然也要使用不同的思路解題,這次就使用迴圈的方式解題吧。

我們每一次迴圈,就搜尋一次當前 i 層是不是有 Left 與 Right 節點為空的節點,如果有就當作是最小深度回傳,如果沒有就將該層丟入 queue 內,在下一層再取出檢查。

  1. 宣告一個 queue ,並且把 root 塞入。
  2. 宣告一個 深度 變數,用於記錄目前深度為多少。
  3. 開始 while 循環,只要 queue 的長度不為 0 ,就一直循環。
  4. 從 queue 取出一個 node ,並且檢查是否左右節點都為空,如果是就返回當前深度,如果不是就將左右兩個節點塞入 queue 中,讓第三步循環下去。

原始碼

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func minDepth(root *TreeNode) int {
    if root == nil {
        return 0
    }
    
    depth := 0
    queue := []*TreeNode{root}
    
    for len(queue) > 0 {
        depth++
        length := len(queue)
        for i := 0; i < length; i++ {
            node := queue[0]
            if node.Left == nil && node.Right == nil {
                return depth
            }
            
            if node.Left != nil {
                queue = append(queue, node.Left)
            }
            
            if node.Right != nil {
                queue = append(queue, node.Right)
            }
            queue = queue[1:]
        }
    }
    
    return depth
}

結尾

你有更好或更簡單的解決方案嗎?

歡迎到我的 Facebook Alan 的筆記本 留言,順手給我個讚吧!你的讚將成為我持續更新的動力,感謝你的閱讀,讓我們一起學習成為更好的自己。