YTUoj_鸣人和佐助

题目描述

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int N=205;
typedef pair<int,int> pii;
int n,m,k;
char g[N][N];
int st[N][N];
int cnt,ans;
int bfs(int a,int b)
{
	queue<pii> q;
	q.push({a,b});
	st[a][b]=1;
	int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
	while(!q.empty())
	{
		pii t=q.front();
		q.pop();
		int x=t.first;
		int y=t.second;
		for(int i=0;i<4;i++)
		{
			int xx=x+dx[i],yy=y+dy[i];
			if(xx<=0||yy<=0||xx>m+1||yy>n+1)continue;
			if(g[xx][yy]=='#'&&k==0)continue;
			if(st[xx][yy]==1)continue;
			st[xx][yy]=1;
			if(g[xx][yy]=='#'){
				cnt++;
				q.push({xx,yy});
			}
			if(g[xx][yy]=='*'){
				cnt++;
				q.push({xx,yy});
			}
			if(g[xx][yy]=='+')return cnt;
		}
	}
	return -1;
}
int main()
{
	cin>>n>>m>>k;
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<n;j++)cin>>g[i][j];
	}
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<n;j++)
		{
			if(g[i][j]=='@')ans=bfs(i,j);
		}
	}
	cout<<ans;
	return 0;
}