Question

554. 砖墙

English Version

题目描述

你的面前有一堵矩形的、由 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:
    cnt = defaultdict(int)
    for row in wall:
      width = 0
      for brick in row[:-1]:
        width += brick
        cnt[width] += 1
    if not cnt:
      return len(wall)
    return len(wall) - cnt[max(cnt, key=cnt.get)]

class Solution {
  public int leastBricks(List<List<Integer>> wall) {
    Map<Integer, Integer> cnt = new HashMap<>();
    for (List<Integer> row : wall) {
      int width = 0;
      for (int i = 0, n = row.size() - 1; i < n; i++) {
        width += row.get(i);
        cnt.merge(width, 1, Integer::sum);
      }
    }
    int max = cnt.values().stream().max(Comparator.naturalOrder()).orElse(0);
    return wall.size() - max;
  }
}

func leastBricks(wall [][]int) int {
  cnt := make(map[int]int)
  for _, row := range wall {
    width := 0
    for _, brick := range row[:len(row)-1] {
      width += brick
      cnt[width]++
    }
  }
  max := 0
  for _, v := range cnt {
    if v > max {
      max = v
    }
  }
  return len(wall) - max
}

/**
 * @param {number[][]} wall
 * @return {number}
 */
var leastBricks = function (wall) {
  const cnt = new Map();
  for (const row of wall) {
    let width = 0;
    for (let i = 0, n = row.length - 1; i < n; ++i) {
      width += row[i];
      cnt.set(width, (cnt.get(width) || 0) + 1);
    }
  }
  let max = 0;
  for (const v of cnt.values()) {
    max = Math.max(max, v);
  }
  return wall.length - max;
};