662. 二叉树最大宽度
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(full binary tree)结构相同,但一些节点为空。
每一层的宽度被定义为两个端点(该层最左和最右的非空节点,两端点间的null节点也计入长度)之间的长度。
示例 1:
输入:
1/ \3 2/ \ \ 5 3 9
输出: 4
解释: 最大值出现在树的第 3 层,宽度为 4 (5,3,null,9)。
示例 2:
输入:
1/ 3 / \ 5 3
输出: 2
解释: 最大值出现在树的第 3 层,宽度为 2 (5,3)。
示例 3:
输入:
1/ \3 2 / 5
输出: 2
解释: 最大值出现在树的第 2 层,宽度为 2 (3,2)。
示例 4:
输入:
1/ \3 2/ \ 5 9 / \
6 7
输出: 8
解释: 最大值出现在树的第 4 层,宽度为 8 (6,null,null,null,null,null,null,7)。
注意: 答案在32位有符号整数的表示范围内。
Code
var ans int
var left map[int]intfunc widthOfBinaryTree(root *TreeNode) int {ans = 0left = make(map[int]int)dfs(root, 0, 0)return ans
}func dfs(root *TreeNode, depth, pos int) {if root == nil {return}if _, ok := left[depth]; !ok {left[depth] = pos}ans = max(ans, pos-left[depth]+1)dfs(root.Left, depth+1, 2*pos)dfs(root.Right, depth+1, 2*pos+1)
}func max(x, y int) int {if x > y {return x}return y
}