HDU 2699 Five in a Row (搜索)

题意： 问下一步是否会赢棋。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>


using namespace std;
const int N = 15;
char map[N+9][N+9];
int dx[]={0,0,-1,1,-1,1,-1,1};
int dy[]={-1,1,0,0,-1,1,1,-1};
bool ok(int x,int y)
{
    if(x<0||x>=N) return false;
    if(y<0||y>=N) return false;
    return true;
}
int s(int x,int y,char K,int dir)
{
    int ans = 0;
    while(1)
    {
        x=x+dx[dir],y=y+dy[dir];
        if(!ok(x,y)) return ans;
        if(map[x][y]==K) ans++;
        else return ans;
    }
}
bool oor(int x,int y,char K)
{
    for(int i=0;i<4;i++)
    {
        if(s(x,y,K,i*2)+s(x,y,K,i*2+1)>=4) return true;
    }
    return false;
}
void solve(char K)
{
   // cout<<K<<endl;
    for(int i=0;i<N;i++)
    for(int j=0;j<N;j++)
    if(map[i][j]=='.')
    {
        if(oor(i,j,K))
        {
            printf("YES\n");
            return ;
        }
    }
    printf("NO\n");return ;
}
int main()
{
    freopen("in.txt","r",stdin);
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        int w=0,b=0;
        for(int i=0;i<N;i++)
        {
            scanf("%s",map[i]);
            for(int j=0;j<N;j++)
            if(map[i][j]=='B') b++;
            else if(map[i][j]=='W') w++;
        }
        if(b==w) solve('B');
        else solve('W');
    }
    return 0;
}