LeetCode刷题之路-每日一题-554. 砖墙

题目描述

---

你的面前有一堵矩形的、由 n 行砖块组成的砖墙。这些砖块高度相同（也就是一个单位高）但是宽度不同。每一行砖块的宽度之和应该相等。

你现在要画一条 自顶向下 的、穿过 最少 砖块的垂线。如果你画的线只是从砖块的边缘经过，就不算穿过这块砖。你不能沿着墙的两个垂直边缘之一画线，这样显然是没有穿过一块砖的。

给你一个二维数组 wall ，该数组包含这堵墙的相关信息。其中，wall[i] 是一个代表从左至右每块砖的宽度的数组。你需要找出怎样画才能使这条线 穿过的砖块数量最少 ，并且返回 穿过的砖块数量 。

输入：wall = [[1,2,2,1],[3,1,2],[1,3,2],[2,4],[3,1,2],[1,3,1,1]]
输出：2

---

前言

说实话，一开始，题目给我看懵了。缕了一缕，大概看懂了。解题突破口是：每一行的前n个元素的和，和相等最多的列，那么拥有最小穿过的墙体。因为如果和不相等，意味着存在穿过的行。
一开始写的时候，想的是直接将原二维数组存储加和之后的数字，但是这样需要按列对行进行遍历，与数据结构反了。看了一眼评论区，Hash表！key为加和之后的值，value为出现的次数。然后遍历hash表，取最大值，用行数减去出现最多的和即为所求。
虽然有思路了，但是在写的时候，还是没写出正确答案，因为这里面还有个条件：对于每一行 i ，sum(wall[i]) 应当是相同的。所以不应该统计最后一个元素。

// <columnSum, rowCount>
        Map<Integer, Integer> columnCross = new HashMap<>();
        for (List<Integer> row: wall) {
            int sum = 0;
            // 最后一块砖不需要计算，因为一定是墙的总宽度
            for (int i=0;i<row.size() - 1;i++) {
                sum += row.get(i);
                columnCross.put(sum, columnCross.getOrDefault(sum, 0) + 1);
            }
        }

        int max = 0;
        for (Integer cross: columnCross.values()) {
            max = cross > max ? cross : max;
        }

        return wall.size() - max ;

进阶

解题成功后，逛下各路大神的解法。还真有不一样的解法：利用有限队列进行列比较。思路是：如果按照列比较前缀和，那就需要将前缀最小的开始进行比较，然后对最小的进行列移动加和（前缀和）再入队。入队之后，可能产生多个一样的前缀和，统计出现的个数，即为出现间隙最多的地方。如此一来，确实能降低空间复杂度。因为是通过列进行比较，所以只需要额外的O(n)（行-墙高）即可。虽然时间复杂度要略有逊色，还要乘上有限队列的时间复杂度logN。