“传染算法DFS“: 求细胞数量

"传染算法DFS"

题目描述

一矩形阵列由数字 0 到 9 组成，数字 1 到 9 代表细胞，细胞的定义为沿细胞数字上下左右若还是细胞数字则为同一细胞，求给定矩形阵列的细胞个数。

输入

第一行两个整数代表矩阵大小 n 和 m。 接下来 n 行，每行一个长度为 m 的只含字符 0 到 9 的字符串，代表这个n×m 的矩阵。

输出

一行一个整数代表细胞个数。

样例输入

4 10
0234500067
1034560500
2045600671
0000000089

样例输出

4

提示

0 < m, n <= 400

分析 

一个声动形象的描述，“传染算法”，
如果一个点没有判定过则对四周进行“传染判定”。

#include <bits/stdc++.h>

using namespace std;

int n,m,ans = 0;
void DFS(int nn,int nm,vector<vector<char>>& dis){
    //一个声动形象的描述，“传染算法”
    //如果没有判定过则对四周进行“传染判定”
    if(nn < 0||nn>=n||nm<0||nm>=m)
        return;
    if(dis[nn][nm] =='0'||dis[nn][nm] == '#')
        return;
    dis[nn][nm] = '#';
    DFS(nn-1,nm,dis);
    DFS(nn+1,nm,dis);
    DFS(nn,nm-1,dis);
    DFS(nn,nm+1,dis);
}
int main()
{
    cin>>n>>m;
    vector<vector<char>> dis;
    for(int i = 0;i<n;i++){
        vector<char> tmp;
        for(int j = 0;j<m;j++){
            char tp;
            cin>>tp;
            tmp.push_back(tp);
        }
        dis.push_back(tmp);
    }
    for(int i = 0;i<n;i++){
        for(int j = 0;j<m;j++){
            if(dis[i][j] == '0'||dis[i][j] == '#')
                continue;
            ans++;
            DFS(i,j,dis);
        }
    }
    cout<<ans<<endl;
    return 0;
}