leet1905-回溯

/** <p>給你兩個 <code>m x n</code> 的二進制矩陣 <code>grid1</code> 和 <code>grid2</code> ,它們只包含 <code>0</code> (表示水域)和 <code>1</code> (表示陸地)。一個 <strong>島嶼</strong> 是由 <strong>四個方向</strong> (水平或者豎直)上相鄰的 <code>1</code> 組成的區域。任何矩陣以外的區域都視爲水域。</p> <p>如果 <code>grid2</code> 的一個島嶼,被 <code>grid1</code> 的一個島嶼 <strong>完全</strong> 包含,也就是說 <code>grid2</code> 中該島嶼的每一個格子都被 <code>grid1</code> 中同一個島嶼完全包含,那麼我們稱 <code>grid2</code> 中的這個島嶼爲 <strong>子島嶼</strong> 。</p> <p>請你返回 <code>grid2</code> 中 <strong>子島嶼</strong> 的 <strong>數目</strong> 。</p> <p> </p> <p><strong>示例 1:</strong></p> <img alt="" src="https://assets.leetcode.com/uploads/2021/06/10/test1.png" style="width: 493px; height: 205px;"> <pre><b>輸入:</b>grid1 = [[1,1,1,0,0],[0,1,1,1,1],[0,0,0,0,0],[1,0,0,0,0],[1,1,0,1,1]], grid2 = [[1,1,1,0,0],[0,0,1,1,1],[0,1,0,0,0],[1,0,1,1,0],[0,1,0,1,0]] <b>輸出:</b>3 <strong>解釋:</strong>如上圖所示,左邊爲 grid1 ,右邊爲 grid2 。 grid2 中標紅的 1 區域是子島嶼,總共有 3 個子島嶼。 </pre> <p><strong>示例 2:</strong></p> <img alt="" src="https://assets.leetcode.com/uploads/2021/06/03/testcasex2.png" style="width: 491px; height: 201px;"> <pre><b>輸入:</b>grid1 = [[1,0,1,0,1],[1,1,1,1,1],[0,0,0,0,0],[1,1,1,1,1],[1,0,1,0,1]], grid2 = [[0,0,0,0,0],[1,1,1,1,1],[0,1,0,1,0],[0,1,0,1,0],[1,0,0,0,1]] <b>輸出:</b>2 <strong>解釋:</strong>如上圖所示,左邊爲 grid1 ,右邊爲 grid2 。 grid2 中標紅的 1 區域是子島嶼,總共有 2 個子島嶼。 </pre> <p> </p> <p><strong>提示:</strong></p> <ul> <li><code>m == grid1.length == grid2.length</code></li> <li><code>n == grid1[i].length == grid2[i].length</code></li> <li><code>1 &lt;= m, n &lt;= 500</code></li> <li><code>grid1[i][j]</code> 和 <code>grid2[i][j]</code> 都要麼是 <code>0</code> 要麼是 <code>1</code> 。</li> </ul> <div><div>Related Topics</div><div><li>深度優先搜索</li><li>廣度優先搜索</li><li>並查集</li><li>數組</li><li>矩陣</li></div></div><br><div><li>👍 63</li><li>👎 0</li></div> */ //leetcode submit region begin(Prohibit modification and deletion) class Solution { public int countSubIslands(int[][] grid1, int[][] grid2) { int m = grid1.length; int n = grid1[0].length; //先淹沒不對應的島嶼 for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (grid1[i][j] == 0 && grid2[i][j] == 1) { dfs(grid2, i, j); } } } //然後正常計數 int res = 0; for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (grid2[i][j] == 1) { res++; dfs(grid2, i, j); } } } return res; } void dfs(int[][] grid, int i, int j) { int m = grid.length; int n = grid[0].length; if (i < 0 || j < 0 || i >= m || j >= n) { return; } //是水 if (grid[i][j] == 0) { return; } grid[i][j] = 0; // 淹沒上下左右的陸地 dfs(grid, i + 1, j); dfs(grid, i, j + 1); dfs(grid, i - 1, j); dfs(grid, i, j - 1); } } //leetcode submit region end(Prohibit modification and deletion)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章