病毒扩散

题目描述

有一种病毒扩散十分凶猛，科学家们为了控制住它，把病毒样本拿到显微镜下仔细研究。

病毒样本是一块n行m列的切片。然而科学家们在观察过程中，发现病毒不仅会扩散，还出现了变异，以便逃过抗生素的识别。我们现在用数字1~9来标识不同的病毒变种。

科学家们想要知道，在这个观察样本中，到底出现了多少个病毒。

注意：病毒不仅会扩散，还会变异，十分狡猾。

输入格式

输入数据包括n+1行，第一行包含两个整数，表示所观测切片的大小n和m。

接下来n行，每行一个长度为m的只包含0到9的方阵，其中1~9代表病毒，不同的数字代表病毒的不同变异类型，0表示未被病毒扩散的区域。

n，m≤100

输出格式

输出仅一行，一个整数，表示病毒的数量。

样例 #1

样例输入 #1

3 4
0000
2022
0126

样例输出 #1

2

这是我的代码

#include <iostream>
using namespace std;

const int N = 110;
int n, m;
int matrix[N][N];
bool st[N][N];

void dfs(int x, int y, int k)
{
    st[x][y] = true;
    int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
    for (int i = 0; i < 4; i ++ )
    {
        int nx = x + dx[i], ny = y + dy[i];
        if (0 <= nx && nx < n && 0 <= ny && ny < m)
        {
            
            if (matrix[nx][ny] == k && !st[nx][ny]) dfs(nx, ny, k);
            if (!matrix[nx][ny] && !st[nx][ny]) dfs(nx, ny, k);
        }
    }
}

int main()
{
    cin >> n >> m;
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < m; j ++ )
            cin >> matrix[i][j];

    int res = 0;
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < m; j ++ )
            if (!st[i][j] && matrix[i][j]) dfs(i, j, matrix[i][j]), res ++ ;

    cout << res << endl;

    return 0;
}

谢谢！！！！！！！