1. 程式人生 > 實用技巧 >LeetCode 102. 二叉樹的層序遍歷

LeetCode 102. 二叉樹的層序遍歷

給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。

 

示例:
二叉樹:[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其層次遍歷結果:

[
  [3],
  [9,20],
  [15,7]
]

class Solution(object):
	def levelOrder(self, root):
		"""
		:type root: TreeNode
		:rtype: List[List[int]]
		"""
		if not root:
			return []
		res = []
		queue = [root]
		while queue:
			# 獲取當前佇列的長度,這個長度相當於 當前這一層的節點個數
			size = len(queue)
			tmp = []
			# 將佇列中的元素都拿出來(也就是獲取這一層的節點),放到臨時list中
			# 如果節點的左/右子樹不為空,也放入佇列中
			for _ in xrange(size):
				r = queue.pop(0)
				tmp.append(r.val)
				if r.left:
					queue.append(r.left)
				if r.right:
					queue.append(r.right)
			# 將臨時list加入最終返回結果中
			res.append(tmp)
		return res

class Solution(object):
	def levelOrder(self, root):
		"""
		:type root: TreeNode
		:rtype: List[List[int]]
		"""
		if not root:
			return []
		res = []
		def dfs(index,r):
			# 假設res是[ [1],[2,3] ], index是3,就再插入一個空list放到res中
			if len(res)<index:
				res.append([])
			#  將當前節點的值加入到res中,index代表當前層,假設index是3,節點值是99
			# res是[ [1],[2,3] [4] ],加入後res就變為 [ [1],[2,3] [4,99] ]
			res[index-1].append(r.val)
			# 遞迴的處理左子樹,右子樹,同時將層數index+1
			if r.left:
				dfs(index+1,r.left)
			if r.right:
				dfs(index+1,r.right)
		dfs(1,root)
		return res