Protect sheep

Protect sheep

题目分析
搜索每只狼所在的位置，将它们用狗围住。

代码

#include<iostream>
using namespace std;
char squ[501][501];
int m,n;
int flag=1;
int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
int main()
{
	cin>>m>>n;
	int x,y,ex,ey;
	for(int i=1;i<=m;i++)
		for(int j=1;j<=n;j++)
			cin>>squ[i][j];
	for(int i=1;i<=m;i++)
		for(int j=1;j<=n;j++)
			if(squ[i][j]=='W')
			{
				x=i;
				y=j;
				for(int k=0;k<4;k++)
				{
				ex=x+dir[k][0];
				ey=y+dir[k][1];
				if(ex>=1&&ey>=1&&ex<=m&&ey<=n)
				{
					if(squ[ex][ey]=='S')
					{
					  flag=0;
					  break;
					}
				    if(squ[ex][ey]=='.')
					squ[ex][ey]='D';
			     }
			   }
			}
			if(flag==0)cout<<"No";
			else 
			{
				cout<<"Yes"<<endl;
				for(int i=1;i<=m;i++)
					{
						if(i!=1)
						cout<<endl;
						for(int j=1;j<=n;j++)
						cout<<squ[i][j];
				}
			}
			return 0;
}