2016年12月24日 星期六

[LeetCode] 102. Binary Tree Level Order Traversal

轉自LeetCode

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
return its level order traversal as:
[
  [3],
  [9,20],
  [15,7]
]
<Solution>

這題是要把在同高度的 node 的值 group 在一起

想法如下
  • 一層一層處理,把同一層的 node,按照從左至右的順序放到 queue
  • 只要 queue 不為空,就代表還沒歷遍完
  • 當下 queue 的 size,就代表有多少個 node 在同一層 level,逐一取出來,並將值放到答案
code 如下
c++

kotlin
這邊再看一個 recursive 的寫法 (參考資料)

想法也是一樣,根據 level 去將每個 node 的值填到對應的 vector 上

code 如下

沒有留言:

張貼留言