跳转至

LeetCode: 554. 砖墙

1、题目描述

你的面前有一堵方形的、由多行砖块组成的砖墙。 这些砖块高度相同但是宽度不同。你现在要画一条自顶向下的、穿过最少砖块的垂线。

砖墙由行的列表表示。 每一行都是一个代表从左至右每块砖的宽度的整数列表。

如果你画的线只是从砖块的边缘经过,就不算穿过这块砖。你需要找出怎样画才能使这条线穿过的砖块数量最少,并且返回穿过的砖块数量。

你不能沿着墙的两个垂直边缘之一画线,这样显然是没有穿过一块砖的。

示例:

输入: [[1,2,2,1],
      [3,1,2],
      [1,3,2],
      [2,4],
      [3,1,2],
      [1,3,1,1]]

输出: 2

解释:

img

提示:

  • 每一行砖块的宽度之和应该相等,并且不能超过 INT_MAX
  • 每一行砖块的数量在 [1,10,000] 范围内, 墙的高度在 [1,10,000] 范围内, 总的砖块数量不超过 20,000

2、解题思路

  • 使用累加值判断
  • 最小的一定是经过某几块转的间隙的,仅此,一行的砖头进行累加,累加值表示缝隙,对每个值放入字典求出现的次数,那么穿越的砖块的次数就是减去砖缝出现的次数
from collections import defaultdict
from itertools import accumulate


class Solution:
    def leastBricks(self, wall: List[List[int]]) -> int:
        total = sum(wall[0])
        max_length = len(wall)
        res = len(wall)
        mapping = defaultdict(int)

        for row in wall:
            for num in accumulate(row):
                if num != total:
                    mapping[num] += 1

        for val in mapping.values():
            res = min(res, max_length - val)
        return res