1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
vertical-order-traversal-of-a-binary-tree.py 954 Bytes
一键复制 编辑 原始数据 按行查看 历史
# Time: O(nlogn)
# Space: O(n)
import collections
# Definition for a binary tree node.
class TreeNode(object):
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution(object):
def verticalTraversal(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
def dfs(node, lookup, x, y):
if not node:
return
lookup[x][y].append(node)
dfs(node.left, lookup, x-1, y+1)
dfs(node.right, lookup, x+1, y+1)
lookup = collections.defaultdict(lambda: collections.defaultdict(list))
dfs(root, lookup, 0, 0)
result = []
for x in sorted(lookup):
report = []
for y in sorted(lookup[x]):
report.extend(sorted(node.val for node in lookup[x][y]))
result.append(report)
return result
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助