砖墙

你的面前有一堵矩形的、由 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):初始化为 “”
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值