代码随想录 101. 孤岛的总面积

101. 孤岛的总面积

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
 
int dfs(vector<vector<int>>& mp, vector<vector<int>>& visit, int y, int x, int s, int& notEdge){
    if (mp[y][x] == 0 || visit[y][x] == 1) return s * notEdge;
    if (mp[y][x] == 1 && visit[y][x] == 0){
        s++;
        visit[y][x] = 1;
    }
    if (y == 0 || y == mp.size() - 1 || x == 0 || x == mp[0].size() - 1) notEdge = 0;
    if (y > 0) s = dfs(mp, visit, y - 1, x, s, notEdge);
    if (x > 0) s = dfs(mp, visit, y, x - 1, s, notEdge);
    if (y < mp.size() - 1) s = dfs(mp, visit, y + 1, x, s, notEdge);
    if (x < mp[0].size() - 1) s = dfs(mp, visit, y, x + 1, s, notEdge);
    return s * notEdge;
}
 
int main(){
    int n, m;
    cin >> n >> m;
    int result = 0;
    vector<vector<int>> mp(n, vector<int>(m, 0));
    vector<vector<int>> visit(n, vector<int>(m, 0));
    for (int i = 0; i < n; i++){
        for (int j = 0; j < m; j++){
            cin >> mp[i][j];
        }
    }
    int isEdge;
    for (int i = 0; i < mp.size(); i++){
        for (int j = 0; j < mp[0].size(); j++){
            if (mp[i][j] == 1 && visit[i][j] == 0){
                isEdge = 1;
                result += dfs(mp, visit, i, j, 0, isEdge);
            }
        }
    }
    cout << result;
    return 0;
}

dfs版本,在之前求总面积的基础上增加了判断是否有有边缘位置元素的部分,所以增加了一个notEdge变量,在遇到边缘元素时标记为0,以使所有输出均变为0,此外需要注意,即使遇到了边缘元素也需要先将所有陆地遍历并在visit上标记。毕竟做人需要有始有终)

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;

int bfs(vector<vector<int>>& mp, vector<vector<int>>& visit, int y, int x){
    int result = 0;
    int notEdge = 1;
    queue<pair<int, int>> q;
    q.push({y, x});
    visit[y][x] = 1;
    result++;
    while(!q.empty()){
        pair<int, int> tmp = q.front();
        q.pop();
        int tmpy = tmp.first, tmpx = tmp.second;
        if (tmpy == 0 || tmpx == 0 || tmpy == mp.size() - 1 || tmpx == mp[0].size() - 1) notEdge = 0;
        
        if (tmpy > 0){
            if (mp[tmpy - 1][tmpx] == 1 && visit[tmpy - 1][tmpx] == 0){
                q.push({tmpy - 1, tmpx});
                visit[tmpy - 1][tmpx] = 1;
                result++;
            }
        }
        if (tmpx > 0){
            if (mp[tmpy][tmpx - 1] == 1 && visit[tmpy][tmpx - 1] == 0){
                q.push({tmpy, tmpx - 1});
                visit[tmpy][tmpx - 1] = 1;
                result++;
            }
        }
        if (tmpy < mp.size() - 1){
            if (mp[tmpy + 1][tmpx] == 1 && visit[tmpy + 1][tmpx] == 0){
                q.push({tmpy + 1, tmpx});
                visit[tmpy + 1][tmpx] = 1;
                result++;
            }
        }
        if (tmpx < mp[0].size() - 1){
            if (mp[tmpy][tmpx + 1] == 1 && visit[tmpy][tmpx + 1] == 0){
                q.push({tmpy, tmpx + 1});
                visit[tmpy][tmpx + 1] = 1;
                result++;
            }
        }
    }
    return result * notEdge;
}

int main(){
    int n, m;
    cin >> n >> m;
    int result = 0;
    vector<vector<int>> mp(n, vector<int>(m, 0));
    vector<vector<int>> visit(n, vector<int>(m, 0));
    for (int i = 0; i < n; i++){
        for (int j = 0; j < m; j++){
            cin >> mp[i][j];
        }
    }
    for (int i = 0; i < mp.size(); i++){
        for (int j = 0; j < mp[0].size(); j++){
            if (mp[i][j] == 1 && visit[i][j] == 0){
                result += bfs(mp, visit, i, j);
            }
        }
    }
    cout << result;
    return 0;
}

bfs的版本,bfs不需要做递归回溯,所以函数的变量还是不变,只需要在内部增加一个标记是否有边缘元素的变量即可。

101. 孤岛的总面积

相关推荐
pianmian139 分钟前
python数据结构基础(7)
数据结构·算法
好奇龙猫3 小时前
【学习AI-相关路程-mnist手写数字分类-win-硬件:windows-自我学习AI-实验步骤-全连接神经网络(BPnetwork)-操作流程(3) 】
人工智能·算法
sp_fyf_20243 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-01
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘
香菜大丸4 小时前
链表的归并排序
数据结构·算法·链表
jrrz08284 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
oliveira-time4 小时前
golang学习2
算法
南宫生5 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
懒惰才能让科技进步5 小时前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
Ni-Guvara6 小时前
函数对象笔记
c++·算法
泉崎6 小时前
11.7比赛总结
数据结构·算法