cs: add «1905. Count Sub Islands»
Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
parent
b907b58d85
commit
4856bc8f3e
1 changed files with 110 additions and 0 deletions
110
cs/count-sub-islands.cs
Normal file
110
cs/count-sub-islands.cs
Normal file
|
@ -0,0 +1,110 @@
|
|||
public class Solution {
|
||||
private static readonly int WATER = 0;
|
||||
private static readonly int ISLAND = 1;
|
||||
|
||||
private static readonly List<(int dy, int dx)> DIRECTIONS = [
|
||||
(1, 0),
|
||||
(-1, 0),
|
||||
(0, 1),
|
||||
(0, -1)
|
||||
];
|
||||
|
||||
private static void RunBFS(int[][] grid, int y0, int x0, int mark) {
|
||||
var q = new Queue<(int y, int x)>();
|
||||
|
||||
// Handle the start
|
||||
grid[y0][x0] = mark;
|
||||
q.Enqueue((y0, x0));
|
||||
|
||||
while (q.TryDequeue(out var next)) {
|
||||
var (y, x) = next;
|
||||
|
||||
foreach (var (dy, dx) in DIRECTIONS) {
|
||||
var nextY = y + dy;
|
||||
var nextX = x + dx;
|
||||
|
||||
if (nextY < 0 || nextY >= grid.Length || nextX < 0 || nextX >= grid[nextY].Length) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (grid[nextY][nextX] != ISLAND) {
|
||||
continue;
|
||||
}
|
||||
|
||||
grid[nextY][nextX] = mark;
|
||||
q.Enqueue((nextY, nextX));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void MarkIslands(int[][] grid) {
|
||||
int mark = 1;
|
||||
|
||||
for (int y = 0; y < grid.Length; ++y) {
|
||||
for (int x = 0; x < grid[y].Length; ++x) {
|
||||
if (grid[y][x] == ISLAND) {
|
||||
RunBFS(grid, y, x, ++mark);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private bool CheckSubIsland(int[][] grid1, int[][] grid2, int y0, int x0) {
|
||||
if (grid1[y0][x0] == WATER || grid2[y0][x0] == WATER) {
|
||||
return false;
|
||||
}
|
||||
|
||||
var q = new Queue<(int y, int x)>();
|
||||
|
||||
// Handle the start; sink the island to optimize for time complexity
|
||||
var expectedMark = grid1[y0][x0];
|
||||
grid2[y0][x0] = WATER;
|
||||
q.Enqueue((y0, x0));
|
||||
|
||||
var isSubIsland = true;
|
||||
while (q.TryDequeue(out var next)) {
|
||||
var (y, x) = next;
|
||||
|
||||
foreach (var (dy, dx) in DIRECTIONS) {
|
||||
var nextY = y + dy;
|
||||
var nextX = x + dx;
|
||||
|
||||
// bounds
|
||||
if (nextY < 0 || nextY >= grid2.Length || nextX < 0 || nextX >= grid2[nextY].Length) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// island continuation
|
||||
if (grid2[nextY][nextX] != ISLAND) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// check if the mark matches
|
||||
if (grid1[nextY][nextX] != expectedMark) {
|
||||
isSubIsland = false;
|
||||
}
|
||||
|
||||
// sink to optimize
|
||||
grid2[nextY][nextX] = WATER;
|
||||
q.Enqueue((nextY, nextX));
|
||||
}
|
||||
}
|
||||
|
||||
return isSubIsland;
|
||||
}
|
||||
|
||||
public int CountSubIslands(int[][] grid1, int[][] grid2) {
|
||||
MarkIslands(grid1);
|
||||
|
||||
int subislands = 0;
|
||||
for (int y = 0; y < grid2.Length; ++y) {
|
||||
for (int x = 0; x < grid2[y].Length; ++x) {
|
||||
if (CheckSubIsland(grid1, grid2, y, x)) {
|
||||
++subislands;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return subislands;
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue