mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-07 15:45:40 +08:00
Merge pull request #2249 from fwqaaq/fwqaaq-patch-1020
Update 1020.飞地的数量.md about rust
This commit is contained in:
@ -491,6 +491,89 @@ class Solution:
|
||||
return ans
|
||||
```
|
||||
|
||||
### Rust
|
||||
|
||||
dfs:
|
||||
|
||||
```rust
|
||||
impl Solution {
|
||||
const DIRECTIONS: [(i32, i32); 4] = [(0, 1), (1, 0), (-1, 0), (0, -1)];
|
||||
|
||||
pub fn num_enclaves(mut grid: Vec<Vec<i32>>) -> i32 {
|
||||
for i in 0..grid.len() {
|
||||
for j in 0..grid[0].len() {
|
||||
if (i == 0 || i == grid.len() - 1 || j == 0 || j == grid[0].len() - 1)
|
||||
&& grid[i][j] == 1
|
||||
{
|
||||
Self::dfs(&mut grid, (i as i32, j as i32));
|
||||
}
|
||||
}
|
||||
}
|
||||
grid.iter()
|
||||
.map(|nums| nums.iter().filter(|&&num| num == 1).count() as i32)
|
||||
.sum()
|
||||
}
|
||||
|
||||
pub fn dfs(grid: &mut [Vec<i32>], (x, y): (i32, i32)) {
|
||||
grid[x as usize][y as usize] = 0;
|
||||
for (dx, dy) in Self::DIRECTIONS {
|
||||
let (nx, ny) = (x + dx, y + dy);
|
||||
if nx < 0 || nx >= grid.len() as i32 || ny < 0 || ny >= grid[0].len() as i32 {
|
||||
continue;
|
||||
}
|
||||
if grid[nx as usize][ny as usize] == 0 {
|
||||
continue;
|
||||
}
|
||||
Self::dfs(grid, (nx, ny));
|
||||
}
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
bfs:
|
||||
|
||||
```rust
|
||||
use std::collections::VecDeque;
|
||||
impl Solution {
|
||||
const DIRECTIONS: [(i32, i32); 4] = [(0, 1), (1, 0), (-1, 0), (0, -1)];
|
||||
|
||||
pub fn num_enclaves(mut grid: Vec<Vec<i32>>) -> i32 {
|
||||
for i in 0..grid.len() {
|
||||
for j in 0..grid[0].len() {
|
||||
if (i == 0 || i == grid.len() - 1 || j == 0 || j == grid[0].len() - 1)
|
||||
&& grid[i][j] == 1
|
||||
{
|
||||
// Self::dfs(&mut grid, (i as i32, j as i32));
|
||||
Self::bfs(&mut grid, (i as i32, j as i32));
|
||||
}
|
||||
}
|
||||
}
|
||||
grid.iter()
|
||||
.map(|nums| nums.iter().filter(|&&num| num == 1).count() as i32)
|
||||
.sum()
|
||||
}
|
||||
|
||||
pub fn bfs(grid: &mut [Vec<i32>], (x, y): (i32, i32)) {
|
||||
let mut queue = VecDeque::new();
|
||||
queue.push_back((x, y));
|
||||
grid[x as usize][y as usize] = 0;
|
||||
while let Some((cur_x, cur_y)) = queue.pop_front() {
|
||||
for (dx, dy) in Self::DIRECTIONS {
|
||||
let (nx, ny) = (cur_x + dx, cur_y + dy);
|
||||
if nx < 0 || nx >= grid.len() as i32 || ny < 0 || ny >= grid[0].len() as i32 {
|
||||
continue;
|
||||
}
|
||||
|
||||
if grid[nx as usize][ny as usize] == 0 {
|
||||
continue;
|
||||
}
|
||||
queue.push_back((nx, ny));
|
||||
grid[nx as usize][ny as usize] = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
|
||||
## 类似题目
|
||||
|
Reference in New Issue
Block a user