[LeetCode]题解(python):102 Binary Tree Level Order Traversal

题目来源


https://leetcode.com/problems/binary-tree-level-order-traversal/

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).


题意分析


Input: 一个二叉树

Output:一个每层数值组合的list

Conditions:层次遍历


题目思路


采用dfs,先序遍历的方法,采用一个level值来记录高度,注意要适时扩充list的size


AC代码(Python)

 1 # Definition for a binary tree node.
 2 # class TreeNode(object):
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.left = None
 6 #         self.right = None
 7 
 8 class Solution(object):
 9     def preorder(self, root, level, res):
10         if root:
11             if len(res) < level + 1: res.append([])
12             res[level].append(root.val)
13             self.preorder(root.left, level + 1, res)
14             self.preorder(root.right, level + 1, res)
15     def levelOrder(self, root):
16         """
17         :type root: TreeNode
18         :rtype: List[List[int]]
19         """
20         res = []
21         self.preorder(root, 0, res)
22         print res
23         return res

 

posted @ 2016-05-17 16:46  loadofleaf  Views(893)  Comments(0Edit  收藏  举报