UVA 657 - The die is cast(cpp file)

Problem Type : Graph (DFS)





#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAX 51
using namespace std;
int X[4]= {-1,0,0,1};
int Y[4]= {0,-1,1,};
char mat[MAX][MAX];
int color[MAX][MAX],dot[MAX][MAX],n,m,cou=0;
void dfs1(int i,int j)
{
    dot[i][j]=1;
    int vx,vy,k;
    for(k=0; k<4; k++)
    {
        vx= i+X[k];
        vy =j+Y[k];
        if(vx>=0 && vx<n && vy>=0 && vy<m)
        {
            if(mat[vx][vy]=='X' && dot[vx][vy]==0)
            {
                dfs1(vx,vy);
            }
        }
    }
}
int dfs(int i,int j)
{
    if(mat[i][j]=='X'&& dot[i][j]==0)
    {
        dfs1(i,j);
        cou++;
    }
    color[i][j]=1;
    int vx,vy,k;
    for(k=0; k<4; k++)
    {
        vx=i+X[k];
        vy=j+Y[k];
        if(vx>=0 && vx<n && vy>=0 && vy<m && (mat[vx][vy]=='*'|| mat[vx][vy]=='X'))
        {
            if(color[vx][vy]==0)
            {
                dfs(vx,vy);
            }
        }
    }

    return cou;
}


int main()
{
    int i,j,k,arr[MAX],cas=0;
    while(scanf("%d %d",&m,&n)&&m&&n)
    {

        cas++;
        for(i=0; i<n; i++)
        {
            for(j=0; j<m; j++)
            {
                scanf(" %c",&mat[i][j]);
            }
        }
        k=0;
        int sp=0;
        int l;
        for(i=0; i<n; i++)
        {
            for(j=0; j<m; j++)
            {
                cou=0;
                if((mat[i][j]=='*' && color[i][j]==0)|| (mat[i][j]=='X' && color[i][j]==0))
                {
                    l=dfs(i,j);
                    arr[k++]=l;
                    sp=1;
                }
            }
        }
        sort(arr,arr+k);
        printf("Throw %d\n",cas);
        if(sp==1)
        {
            for(i=0; i<k-1; i++)
            {
                printf("%d ",arr[i]);
            }
            printf("%d",arr[k-1]);
        }
        else
        {
            printf("0");
        }
        printf("\n\n");
        memset(arr,0,sizeof(arr));
        memset(color,0,sizeof(color));
        memset(dot,0,sizeof(dot));
        memset(mat,'\0',sizeof(mat));
    }
}