华为面试题： 砌墙

链接

题目

你的面前有一堵矩形的、由 n 行砖块组成的砖墙。这些砖块高度相同（也就是一个单位高）但是宽度不同。每一行砖块的宽度之和相等。
你现在要画一条 自顶向下 的、穿过 最少 砖块的垂线。如果你画的线只是从砖块的边缘经过，就不算穿过这块砖。你不能沿着墙的两个垂直边缘之一画线，这样显然是没有穿过一块砖的。
给你一个二维数组 wall ，该数组包含这堵墙的相关信息。其中，wall[i] 是一个代表从左至右每块砖的宽度的数组。你需要找出怎样画才能使这条线 穿过的砖块数量最少 ，并且返回 穿过的砖块数量 。

示例 1：6

输入：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 <= 10^4
1 <= wall[i].length <= 10^4
1 <=sum(wall[i].length) <= 2 * 10^4
对于每一行 i ，sum(wall[i]) 是相同的
1 <=wall[i][j] <= 2^31 - 1

代码

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

注意点与分析

分析：每一层的前缀和统计
由于是前缀和，且元素大小>1，所以不会出现重复元素。

此处注意代码：

            for(int i = 0; i < n - 1; ++i){
                sum += level[i];
                cap[sum]++;
            }

此处的i < n-1是表示不统计最后一个元素的。因为每一层的砖的总长是一样的！！

统计某一处前缀和在每层出现的最多，就是答案。

由于每一层层内不出现重复元素，建立映射map，将前缀和和出现次数做映射统计，求出其中出现次数最多的即可！