LeetCode 554. 砖墙（map计数）

1. 题目

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

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

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

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

在这里插入图片描述

示例：
输入: [[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。

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/brick-wall
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

前缀和记录位置坐标pos的计数，存入哈希map
取每个pos出来，找出pos位置计数最大的，穿过的砖块最少

class Solution {	// c++
public:int leastBricks(vector<vector<int>>& wall) {int pos = 0, n = wall.size(), minbrick = INT_MAX;unordered_map<int,int> m;//pos，计数for(int i = 0, j; i < n; i++) {pos = 0;for(j = 0; j < wall[i].size()-1; j++){pos += wall[i][j];m[pos]++;}}for(auto it = m.begin(); it != m.end(); ++it)minbrick = min(minbrick, n-it->second);return minbrick==INT_MAX ? n : minbrick;}
};

96 ms 19.3 MB

class Solution: # py3def leastBricks(self, wall: List[List[int]]) -> int:pos = 0n = len(wall)minbrick = float("inf")dict = {}for i in range(n):pos = 0for j in range(len(wall[i])-1):pos += wall[i][j]if pos not in dict:dict[pos] = 1else:dict[pos] += 1for p in dict:minbrick = min(minbrick, n-dict[p])if minbrick==float("inf"):return nreturn minbrick