Leetcode 554.砖墙
1 题目描述(Leetcode题目链接)
你的面前有一堵矩形的、由多行砖块组成的砖墙。 这些砖块高度相同但是宽度不同。你现在要画一条自顶向下的、穿过最少砖块的垂线。
砖墙由行的列表表示。 每一行都是一个代表从左至右每块砖的宽度的整数列表。
如果你画的线只是从砖块的边缘经过,就不算穿过这块砖。你需要找出怎样画才能使这条线穿过的砖块数量最少,并且返回穿过的砖块数量。
你不能沿着墙的两个垂直边缘之一画线,这样显然是没有穿过一块砖的。
输入: [[1,2,2,1],
[3,1,2],
[1,3,2],
[2,4],
[3,1,2],
[1,3,1,1]]
输出: 2
提示:
- 每一行砖块的宽度之和应该相等,并且不能超过 INT_MAX。
- 每一行砖块的数量在 [1,10,000] 范围内, 墙的高度在 [1,10,000] 范围内, 总的砖块数量不超过 20,000。
2 题解
记录每个数组的前缀和出现次数,画线就应该在前缀和出现次数最多的位置画线。例如题目样例中,前缀和4出现了4次,因此画线就在这个位置画。
class Solution:
def leastBricks(self, wall: List[List[int]]) -> int:
m = len(wall)
max_val = 0
dic = collections.defaultdict(int)
for i in range(m):
s = 0
for j in range(len(wall[i]) - 1):
s += wall[i][j]
dic[s] += 1
max_val = max(max_val, dic[s])
return m - max_val