题意: 问下一步是否会赢棋。
#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;
}