你的面前有一堵矩形的、由 n 行砖块组成的砖墙。这些砖块高度相同(也就是一个单位高)但是宽度不同。每一行砖块的宽度之和应该相等。
你现在要画一条 自顶向下 的、穿过 最少 砖块的垂线。如果你画的线只是从砖块的边缘经过,就不算穿过这块砖。你不能沿着墙的两个垂直边缘之一画线,这样显然是没有穿过一块砖的。
给你一个二维数组 wall ,该数组包含这堵墙的相关信息。其中,wall[i] 是一个代表从左至右每块砖的宽度的数组。你需要找出怎样画才能使这条线 穿过的砖块数量最少 ,并且返回 穿过的砖块数量 。
看到这个题刚开始想的是使用一个列表,长度初始化为每行砖墙的宽度之和,每个元素初始化为砖墙的总厚度,即:
rows, cols = len(wall), sum(wall[0])
res = [rows] * cols
然后开始遍历,对每行的砖宽度进行累加,在砖缝位置 -1 ,最后返回最小厚度, 代码如下:
class Solution:
def leastBricks(self, wall: List[List[int]]) -> int:
rows, cols = len(wall), sum(wall[0])
res = [rows] * cols
for i in range(rows):
nowall = 0
for w in wall[i]:
nowall += w
res[nowall-1] -= 1
res[-1] = rows
return min(res)
理想是丰满的,现实是残酷的。最后被测试用例 [[100000000],[100000000],[100000000]] 给搞超时了。。。可以看出,砖墙总宽度越宽,我这耗时就会更长,应该主要是查找最小值那里耗时最多。
因此需要采用另一种数据结构,减少耗时,于是想到了哈希表:
class Solution:
from collections import defaultdict
def leastBricks(self, wall: List[List[int]]) -> int:
rows = len(wall)
hashmap = defaultdict(int)
for i in range(rows):
nowall = 0
for j in range(0, len(wall[i])-1):
nowall += wall[i][j]
hashmap[nowall] += 1
return rows - max(hashmap.values(), default=0)
模块collections是数据类型容器模块,defaultdict(function_factory) 构建的是一个类似 dictionary 的对象,其中 keys 的值,自行确定赋值,但是 values 的类型,是 function_factory 的类实例,而且具有默认值。
- defaultdict(int):初始化为 0
- defaultdict(float):初始化为 0.0
- defaultdict(str):初始化为 “”