要求
你的面前有一堵矩形的、由 n 行砖块组成的砖墙。这些砖块高度相同(也就是一个单位高)但是宽度不同。每一行砖块的宽度之和相等。
你现在要画一条 自顶向下 的、穿过 最少 砖块的垂线。如果你画的线只是从砖块的边缘经过,就不算穿过这块砖。你不能沿着墙的两个垂直边缘之一画线,这样显然是没有穿过一块砖的。
给你一个二维数组 wall ,该数组包含这堵墙的相关信息。其中,wall[i] 是一个代表从左至右每块砖的宽度的数组。你需要找出怎样画才能使这条线 穿过的砖块数量最少 ,并且返回 穿过的砖块数量 。
示例 1:
输入:wall = [[1,2,2,1],[3,1,2],[1,3,2],[2,4],[3,1,2],[1,3,1,1]]
输出:2
示例 2:
输入:wall = [[1],[1],[1]]
输出:3
提示:
- n == wall.length
- 1 <= n <= 104
- 1 <= wall[i].length <= 104
- 1 <= sum(wall[i].length) <= 2 * 104
- 对于每一行 i ,sum(wall[i]) 是相同的
- 1 <= wall[i][j] <= 231 - 1
核心代码
class Solution:
def leastBricks(self, wall: List[List[int]]) -> int:
mark = {}
for x in wall:
tmp = 0
for i in range(len(x) - 1):
tmp += x[i]
mark[tmp] = mark.get(tmp,0) + 1
if not mark:
return len(wall)
return len(wall) - max(mark.values())
解题思路:这题极端情况,一条垂线穿过最多砖块的数目就是,整个砖墙的行数len(walls)。那么我们对每个位置都 建立index与砖块边缘数目的关系。比如题目中给的例子:行数=len(walls)=6,列数=6,对于index=0和index=6是整个砖墙的边缘,不考虑。对于index=1时,在该位置结束的砖块数有3个,index=2时,在该位置结束的砖块数仅有1个,index=3时,在该位置结束的砖块数有3个,index=4时,在该位置结束的砖块数有4个,是最多的情况,所以len(walls)-4,就是本题的答案
注意:记录index与数目关系时,我一开始考虑用list,初始化list中每个元素为0,但发现没有用字典好,字典可以不初始化,对每个dict[key]自增1时,如果该key原来没有值,就返回0。