读前福利:几百本互联网技术书籍送给大家https://mp.weixin.qq.com/s/dFqVQ2qJxvQ0YrIlPISJuw

每天分享一个LeetCode题目
每天 5 分钟,一起进步
LeetCode N 叉树的前序遍历,地址: https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal/
树结点类
1 2 3 4
| class TreeNode(object): def __init__(self, val, children=[]): self.val = val self.children = children
|
N 叉树的前序遍历
利用递归,依然遵循「跟左右」的遍历原则
1 2 3 4
| def preorder(self, root): print(root.val, end=" ") for node in root.children: self.preorder(node)
|
是不是看起来特别简单,但是这样不符合 LeetCode 题目中的要求
需要将结果放到一个数组中,所以需要提前初始化一个 list 进行存放
重新编码看看
1 2 3 4 5 6 7 8 9 10 11 12 13 14
| def preorder(self, root):
res = [] def dfs(root): if not root: return res.append(root.val) for node in root.children: dfs(node) dfs(root) return res
|
就是提前初始化了 res,然后在遍历的时候进赋值操作
完整代码
可直接执行
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
|
class Node(object): def __init__(self, val=None, children=[]): self.val = val self.children = children
class Solution(object): def preorder(self, root):
res = [] def dfs(root): if not root: return res.append(root.val) for node in root.children: dfs(node) dfs(root) return res
if __name__ == "__main__": root = Node('A') node_B = Node('B') node_C = Node('C') node_D = Node('D') node_E = Node('E') node_F = Node('F') node_G = Node('G') node_H = Node('H') node_I = Node('I') root.children = [node_B, node_C, node_D] node_B.children = [node_E, node_F, node_G] node_D.children = [node_H, node_I]
s = Solution() print(s.preorder(root))
|
最后
在这里给大家准备了几百本的互联网技术类书籍,需要的来下载吧!点击获取
有任何问题,欢迎随时交流!!!