zzuli 1732: 五子棋 (搜索+模拟)

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
#include<vector>
using namespace std;
#define mem(a,t) memset(a,t,sizeof(a))
#define N 115
#define ll __int64
const int inf=0x1f1f1f1f;
const double eps=1e-7;
char s[N][N];
int dir[8][2]={1,1,1,0,1,-1,0,-1,-1,-1,-1,0,-1,1,0,1};
int n,m;
int judge(int x,int y) 
{
    if(x>=0&&x<n&&y>=0&&y<m)
        return 1;
    return 0;
}

int bfs(int f)      //判断f方是否存在5子连续
{
    char tmp=f+'0';
    int i,j,k,cnt,x,y;
    for(i=0;i<n;i++)
    {
        for(j=0;j<m;j++)
        {
            if(s[i][j]==tmp)
            {
                for(k=0;k<8;k++)
                {
                    x=i;
                    y=j;
                    cnt=1;
                    while(1)
                    {
                        x+=dir[k][0];
                        y+=dir[k][1];
                        if(!judge(x,y)||s[x][y]!=tmp)
                            break;
                        cnt++;
                    }
                    if(cnt>=5)
                        return 1;
                }
            }
        }
    }
    return 0;
}

void fun(int f)    //去掉一子
{
    int i,j,flag=0;
    char tmp=f+'0';
    for(i=0;i<n;i++)
    {
        for(j=0;j<m;j++)
        {
            if(s[i][j]==tmp)
            {
                s[i][j]='0';
                if(bfs(f)==0)
                {
                    flag=1;
                    break;
                }
                s[i][j]=tmp;
            }
        }
        if(flag)
            break;
    }
    if(flag)
    {
        if(f==1)
            printf("white\n");
        else
            printf("black\n");
    }
    else
        printf("fault\n");
}

int main()
{
    int i,j,T;
    int b,w;
    int f1,f2;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        b=w=0;
        for(i=0;i<n;i++)
        {
            scanf("%s",s[i]);
            for(j=0;j<m;j++)
            {
                if(s[i][j]=='1')
                    w++;
                else if(s[i][j]=='2')
                    b++;
            }
        }
        if(b!=w&&b!=w+1)
        {
            printf("fault\n");
            continue;
        }
        f1=bfs(1);
        f2=bfs(2);
        if(f1&&f2)
        {
            printf("fault\n");
            continue;
        }
        if(f1)
        {
            if(w!=b)
                printf("fault\n");
            else
                fun(1);
        }
        else if(f2)
        {
            if(b!=w+1)
                printf("fault\n");
            else
                fun(2);
        }
        else
            printf("other\n");
    }
    return 0;
}