LightOJ 1012 - Guilty Prince (cpp file)

Problem Type : BFS/DFS


#include<stdio.h>
#include<string.h>
#define MAX 21
int v,e,cou=0;
int X[4]= {-1,0,0,1};
int Y[4]= {0,-1,1,0};
char mat[MAX][MAX];
int color[MAX][MAX];
void dfs(int i,int j)
{
    color[i][j]=1;

    int vx,vy,k;
    for(k=0; k<4; k++)
    {
        vx=X[k]+i;
        vy=Y[k]+j;
        if(vx>=0 && vx<v && vy>=0 && vy<e)
        {
            if(color[vx][vy]==0 && mat[vx][vy]=='.')
            {
                cou++;
                dfs(vx,vy);
            }
        }
    }
}
int main()
{
    int test,i,j,cas=0;
    scanf("%d",&test);
    while(test--)
    {
        cas++;
        scanf("%d %d",&e,&v);
        for(i=0; i<v; i++)
        {
            for(j=0; j<e; j++)
            {
                scanf(" %c",&mat[i][j]);
            }
        }
        memset(color,0,sizeof(color));
        cou=1;
        for(i=0; i<v; i++)
        {
            for(j=0; j<e; j++)
            {
                if(mat[i][j]=='@')
                {
                    dfs(i,j);
                }
            }

        }
        printf("Case %d: %d\n",cas,cou);
    }

}