[LeetCode] 102. Binary Tree Level Order Traversal
Posted aaronliu1991
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[LeetCode] 102. Binary Tree Level Order Traversal相关的知识,希望对你有一定的参考价值。
二叉树层序遍历。题干既是题意。如下例子,
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] ]
此题比较简单的做法是用层序遍历BFS,但是此题也可以用DFS深度遍历做,比较巧妙。两种思路的时间和空间复杂度都是O(n)。
BFS
1 /** 2 * @param {TreeNode} root 3 * @return {number[][]} 4 */ 5 var levelOrder = function(root) { 6 let res = []; 7 if (root === null) return res; 8 9 // normal case 10 let queue = [root]; 11 while (queue.length) { 12 let list = []; 13 let size = queue.length; 14 for (let i = 0; i < size; i++) { 15 let cur = queue.shift(); 16 list.push(cur.val); 17 if (cur.left !== null) queue.push(cur.left); 18 if (cur.right !== null) queue.push(cur.right); 19 } 20 res.push(list); 21 } 22 return res; 23 };
DFS
DFS的思路的要点在于,需要在递归函数中多一个参数level,记录递归到树的第几层了。跑一下例子好了。当第一次把根节点3放进res之后,下面就开始遍历他的两个孩子。遍历到左孩子9的时候,level是1,大于等于res.length(1),所以需要再加入一个subarray(15行)以便于加入9这个节点(17行)。当遍历右孩子20的时候,level依然是1,并不大于等于res.length(2),所以此时并不需要再加入subarray了。但是20依然可以被放进最后的结果集。简而言之,DFS用了一个level参数来判断是否大于结果集此时的长度,以决定是否需要再添加新的subarray来存放下一层的节点值。
Given binary tree
[3,9,20,null,null,15,7]
,3 / 9 20 / 15 7
1 /** 2 * @param {TreeNode} root 3 * @return {number[][]} 4 */ 5 var levelOrder = function(root) { 6 let res = []; 7 if (root === null) return res; 8 helper(res, root, 0); 9 return res; 10 }; 11 12 var helper = function(res, root, level) { 13 if (root === null) return; 14 if (level >= res.length) { 15 res.push([]); 16 } 17 res[level].push(root.val); 18 helper(res, root.left, level + 1); 19 helper(res, root.right, level + 1); 20 };
以上是关于[LeetCode] 102. Binary Tree Level Order Traversal的主要内容,如果未能解决你的问题,请参考以下文章
LeetCode102. Binary Tree Level Order Traversal
LeetCode 102. Binary Tree Level Order Traversal
[LeetCode]题解(python):102- Binary Tree Level Order Traversal
Java [Leetcode 102]Binary Tree Level Order Traversal