挑战任务
某地区爆发病毒,给定一个矩阵,x代表隔离墙,o代表病毒,病毒区域是由它水平方向或垂直方向上相邻的病毒连接而成,斜角连接不会构成病毒区域。你要计算出矩阵中有几个病毒区域(假设矩阵周边是被隔离墙包围的)。
编程要求
请在右侧编辑器中填充代码,补全virusArea(int n,int m,vector<vector >& area)函数,对输入的矩阵进行计算,返回“病毒区域”的个数,函数参数说明如下:
n:矩阵的行
m:矩阵的列
area:待计算的矩阵
测试说明
样例1:
输入:
4 5
x o x x x
x x x o x
x x o o x
x x x x x
输出:2
样例2:
输入:
4 4
o o x x
x x x o
o o o x
x o x o
输出:4
样例3:
输入:
4 6
x o x o x x
x x x o o x
x o x x x x
x o x o o x
输出:4
题目链接:https://www.educoder.net/tasks/53tynao7j8hf
ac代码如下:
#ifndef _TEST
#define _TEST
#include <iostream>
#include <vector>
using namespace std;
int tx[]={-1,1,0,0};
int ty[]={0,0,-1,1};
class Task{
public:
void dfs(int x,int y,vector<vector<char> >& arr,int rx,int ry) //dfs算法
{
int xx,yy;
for(int i=0;i<4;i++) //上下左右四个方向遍历
{
xx=x+tx[i];yy=y+ty[i];
if(xx>=0&&yy>=0&&xx<arr.size()&&yy<arr[xx].size()&&arr[xx][yy]=='o'&&!(xx==rx&&yy==ry)) //坐标存在且不是根节点
{
arr[xx][yy]='x';
dfs(xx,yy,arr,rx,ry); //接着遍历
}
}
}
int virusArea(int n,int m,vector<vector<char> >& arr){
/********* Begin *********/
int ans=0; //答案
for(int i=0;i<n;i++) //逐个遍历病毒区域
for(int j=0;j<m;j++)
if(arr[i][j]=='o')
dfs(i,j,arr,i,j);
for(int i=0;i<n;i++) //搜索答案个数
for(int j=0;j<m;j++)
if(arr[i][j]=='o')
ans++;
return ans; //返回答案
/********* End *********/
}
};
#endif
PS:经典的DFS题目