HDOJ1312 DFS

求一个连通块的元素个数（包含@的连通块）

#include <iostream>
#include"sstream"
#include"string"
#include"algorithm"
#include"cstring"
#include"cstdio"
using namespace std;
int dir[4][2]={1,0,-1,0,0,1,0,-1};
char g[30][30];
int r,c;
int cnd;
int visit[30][30];
void dfs(int si,int sj)
{
for(int i=0;i<4;i++)
{
int ki=si+dir[i][0];
int kj=sj+dir[i][1];
if((ki<r&&ki>=0&&kj<c&&kj>=0)&&(g[ki][kj]=='.'||g[ki][kj]=='@')&&!visit[ki][kj])
{
++cnd;
visit[ki][kj]=1;
dfs(ki,kj);
}

}
}
int main()
{
// freopen("a.txt","r",stdin);
while(cin>>c>>r&&(r+c))
{
memset(visit,0,sizeof(visit));
int si,sj;
int i,j;
cnd=0;
for(i=0;i<r;i++)
for(j=0;j<c;j++)
{
cin>>g[i][j];
if(g[i][j]=='@')
{sj=j;si=i;}
}
++cnd;
visit[si][sj]=1;
dfs(si,sj);
cout<<cnd<<endl;
}
return 0;
}