803. 打砖块

2021-05-22 11:03:36 阅读：215 来源： 互联网

难度 hard
有一个 m x n 的二元网格，其中 1 表示砖块，0 表示空白。砖块稳定（不会掉落）的前提是：

一块砖直接连接到网格的顶部，或者
至少有一块相邻（4 个方向之一）砖块稳定不会掉落时
给你一个数组 hits ，这是需要依次消除砖块的位置。每当消除 hits[i] = (rowi, coli) 位置上的砖块时，对应位置的砖块（若存在）会消失，然后其他的砖块可能因为这一消除操作而掉落。一旦砖块掉落，它会立即从网格中消失（即，它不会落在其他稳定的砖块上）。

返回一个数组 result ，其中 result[i] 表示第 i 次消除操作对应掉落的砖块数目。

注意，消除可能指向是没有砖块的空白位置，如果发生这种情况，则没有砖块掉落。

示例 1：

输入：grid = [[1,0,0,0],[1,1,1,0]], hits = [[1,0]]
输出：[2]
解释：
网格开始为：
[[1,0,0,0]，
[1,1,1,0]]
消除 (1,0) 处加粗的砖块，得到网格：
[[1,0,0,0]
[0,1,1,0]]
两个加粗的砖不再稳定，因为它们不再与顶部相连，也不再与另一个稳定的砖相邻，因此它们将掉落。得到网格：
[[1,0,0,0],
[0,0,0,0]]
因此，结果为 [2] 。
示例 2：

输入：grid = [[1,0,0,0],[1,1,0,0]], hits = [[1,1],[1,0]]
输出：[0,0]
解释：
网格开始为：
[[1,0,0,0],
[1,1,0,0]]
消除 (1,1) 处加粗的砖块，得到网格：
[[1,0,0,0],
[1,0,0,0]]
剩下的砖都很稳定，所以不会掉落。网格保持不变：
[[1,0,0,0],
[1,0,0,0]]
接下来消除 (1,0) 处加粗的砖块，得到网格：
[[1,0,0,0],
[0,0,0,0]]
剩下的砖块仍然是稳定的，所以不会有砖块掉落。
因此，结果为 [0,0] 。

提示：

m == grid.length
n == grid[i].length
1 <= m, n <= 200
grid[i][j] 为 0 或 1
1 <= hits.length <= 4 * 104
hits[i].length == 2
0 <= xi <= m - 1
0 <= yi <= n - 1
所有 (xi, yi) 互不相同

解题思路：这道题目是看了grandyang的解法后，才解决的，思路就是，用一个HashSet来保存稳定的砖块，以grid[i][j]为例，保存对应的索引i*n+j，（n是列数），然后用set的size来更新结果。我们直接的思路就是，从第一行的砖块深度遍历来计算砖块的数量，然后去掉某个砖块后，再次从第一行的砖块深度遍历来计算砖块数量，这样两次遍历完后，砖块数量的差值就是去掉一块转后掉落的砖块，但这种搜索方法就不是很高效。我们可以尝试这种方法，首先把所有需要去掉的砖块都去掉，这样先遍历一遍，得到最后剩下的砖块数量，然后从最后一块砖头开始往回加，每加一块砖头，我们就用深度遍历把和他相连的砖头加入hashset中，增加的砖头就是原来消除这块砖头之后会掉的砖头。具体地，我们先把hits对应的需要消除的位置减一，然后统计依次最后剩下砖头的数量，然后从hits的最后一个位置开始往回加，如果加完不是一，说明原来的位置是0，直接跳过，如果是一，那就看周边有没有稳定的砖块存在（也就是在set中）或者是否是第0行，如果是的话，那就对当前元素进行深度遍历，结束后用set的前后size的差异减去一，就是掉落的砖头数，减一是因为去掉的砖头不算在掉落的砖头中。

代码 t31 s23 java

class Solution {
    public int[] hitBricks(int[][] grid, int[][] hits) {
        HashSet<Integer> set = new HashSet<>();
        int m = grid.length, n = grid[0].length;
        int len = hits.length;
        int[] res = new int[len];
        for(int i=0; i<len; i++){
            grid[hits[i][0]][hits[i][1]] -= 1;
        }
        for(int i=0; i<grid[0].length; i++){
            helper(grid, 0, i, set);
        }
        for(int i=len-1; i>=0; i--){
            int oldSize = set.size(), x = hits[i][0], y = hits[i][1];
            if(++grid[x][y]!=1) continue;
            else{
                if((x+1<m && set.contains((x+1)*n+y)) ||
                (x-1>=0 && set.contains((x-1)*n+y)) ||
                (y+1<n && set.contains(x*n+y+1)) ||
                (y-1>=0 && set.contains(x*n+y-1)) ||
                x==0){
                    helper(grid, x, y, set);
                    res[i] = set.size() - oldSize - 1;
                } 
            }
        }
        return res;
    }

    public void helper(int[][]grid, int i, int j, HashSet<Integer> set){
        int m = grid.length, n = grid[0].length;
        int t = i * n + j;
        if(i<0 || i>=m || j<0 || j>=n || grid[i][j]!=1 || set.contains(t))  return;
        set.add(t);
        helper(grid, i+1, j, set);
        helper(grid, i, j+1, set);
        helper(grid, i-1, j, set);
        helper(grid, i, j-1, set);
    }
}

参考资料:
https://www.cnblogs.com/grandyang/p/9362777.html

标签：hits,set,掉落,int,grid,砖块,803
来源： https://www.cnblogs.com/zhengxch/p/14798316.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

803. 打砖块