我们想要使用一棵四叉树来储存一个 N x N
的布尔值网络。网络中每一格的值只会是真或假。树的根结点代表整个网络。对于每个结点, 它将被分等成四个孩子结点直到这个区域内的值都是相同的.
每个结点还有另外两个布尔变量: isLeaf
和 val
。isLeaf
当这个节点是一个叶子结点时为真。val
变量储存叶子结点所代表的区域的值。
你的任务是使用一个四叉树表示给定的网络。下面的例子将有助于你理解这个问题:
给定下面这个8 x 8
网络,我们将这样建立一个对应的四叉树:
由上文的定义,它能被这样分割:
对应的四叉树应该像下面这样,每个结点由一对 (isLeaf, val)
所代表.
对于非叶子结点,val
可以是任意的,所以使用 *
代替。
提示:
N
将小于1000
且确保是 2 的整次幂。- 如果你想了解更多关于四叉树的知识,你可以参考这个 wiki 页面。
思路:
看到这道题,似乎很难的样子,不过仔细分析的话,还是有规律可循。构建一棵树显然需要用到递归的方法,先是根节点,然后是四个子结点。
重难点:
- 新建函数 allValueSame(grid),判断网格所有值是否相等。这是判断网格是否为叶子结点的依据。
- 若grid只有一个元素,那么必然为叶子结点,且此节点的val根据这元素的1/0来赋值True/False。
- 若grid所有值相等,同2。
- 若grid存在不相等的元素,那么此节点就不是叶子节点,需要把grid分成四份,分别递归调用四次。
- 导入numpy.array,来完成二维数组(即网格)的切片操作。
网格的切片:先将 list 转换为 array,然后使用[ : ,: ]来切片,最后再转换成列表 .tolist()
"""
# Definition for a QuadTree node.
class Node:
def __init__(self, val, isLeaf, topLeft, topRight, bottomLeft, bottomRight):
self.val = val
self.isLeaf = isLeaf
self.topLeft = topLeft
self.topRight = topRight
self.bottomLeft = bottomLeft
self.bottomRight = bottomRight
"""
from numpy import array
class Solution:
def construct(self, grid):
"""
:type grid: List[List[int]]
:rtype: Node
"""
root = Node('*', True, None, None, None, None);
if len(grid) == 1:
root.isLeaf = True;
root.val = True if grid[0][0] == 1 else False;
if self.allValueSame(grid): # 所有值相等
root.isLeaf = True;
root.val = True if grid[0][0] == 1 else False;
else: # 并非所有值相等
halfLength = len(grid) // 2; # 使用 // 表示整除
root.isLeaf = False; # 如果网格中有值不相等,这个节点就不是叶子节点
# 使用array来完成二维数组的切片
root.topLeft = self.construct(array(grid)[:halfLength, :halfLength].tolist());
root.topRight = self.construct(array(grid)[:halfLength, halfLength:].tolist());
root.bottomLeft = self.construct(array(grid)[halfLength:, :halfLength].tolist());
root.bottomRight = self.construct(array(grid)[halfLength:, halfLength:].tolist());
return root;
def allValueSame(self, grid):
"""
:type grid: List[List[int]]
:rtype: boolean
"""
for i in range(len(grid)):
for j in range(len(grid[0])):
if grid[0][0] != grid[i][j]:
return False;
return True;