哈希表554. 砖墙

Posted 2021-06-11 7TribeZ

目录

Python collection模块

哈希表

代码

---

 

554. 砖墙

难度中等

你的面前有一堵矩形的、由 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

 

Python collection模块

https://www.jianshu.com/p/8acb163dd1fe

 

哈希表

题目要求穿过的砖块数量最少，等效于通过的间隙最多。

我们可以使用「哈希表」记录每个间隙的出现次数，最终统计所有行中哪些间隙出现得最多，使用「总行数」减去「间隙出现的最多次数」即是答案。

如何记录间隙呢？直接使用行前缀记录即可。

就用示例数据来举 🌰 ：

第 1 行的间隙有 [1,3,5]
第 2 行的间隙有 [3,4]
第 3 行的间隙有 [1,4]
第 4 行的间隙有 [2]
第 5 行的间隙有 [3,4]
第 6 行的间隙有 [1,4,5]
对间隙计数完成后，遍历「哈希表」找出出现次数最多间隙 4，根据同一个间隙编号只会在单行内被统计一次，用总行数减去出现次数，即得到「最少穿过的砖块数」。

 

代码

c++

class Solution {
public:
    int leastBricks(vector<vector<int>>& wall) {
        unordered_map<int,int> cnt;
        for (auto& widths:wall)
        {
            int n=widths.size();
            int sum=0;
            for (int i=0;i<n-1;i++)
            {
                sum+=widths[i];
                cnt[sum]++;
            }
        }
        int maxCnt=0;
        for (auto& [_, c] :cnt){
            maxCnt=max(maxCnt,c);
        }
        return wall.size()-maxCnt;

    }
};

python

class Solution(object):
    def leastBricks(self, wall):
        """
        :type wall: List[List[int]]
        :rtype: int
        """
        from collections import defaultdict
        time_dct = defaultdict(int)
        max_time = 0
        for r in range(len(wall)):
            sum_ = 0
            for brick in wall[r][:-1]:
                sum_ += brick
                time_dct[sum_] += 1
                max_time = max(time_dct[sum_],max_time)
        return len(wall) - max_time