牛客网 E-对弈

链接：https://www.nowcoder.com/acm/contest/136/E
来源：牛客网

题目描述

  善弈者谋势，不善弈者谋子。                                             ——《弈林新编》               蒟蒻HtBest与神犇WHZ下棋（五子棋），HtBest执黑棋，WHZ执白棋。由于HtBest天资愚笨，不会判断输赢，所以需要你帮他开发一个判断五子棋输赢的程序。      

输入描述:
第一行有2个正整数n，m，分别表示棋盘大小（n*n）和对弈步数。接下来m行，每行两个正整数xi,yi ，表示对弈者下棋的坐标，第2、4、6…行是HtBest下的棋子，第3、5、7…行是WHZ下的棋子。
输出描述:
第一行一个字符串s 和一个正整数num ，用空格隔开，分别表示对弈的胜负结果和该结果出现时的步数，如果HtBest胜则s=“HtBest”，num为HtBest胜利时的步数（为偶数），如果WHZ胜则s=“WHZ”，num为WHZ胜利时的步数（为奇数），如果对弈m步后胜负仍未定，则s=“UNK”，num=m。提示：当棋盘上第一次有五个子连续排列（横竖斜都可）时，胜负已定。在这之后，两人有可能仍继续落子。

简单模拟题，简单dfs

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <ctime>
#include <random>
#include <cstdlib>
#include <algorithm>
using namespace std;

const int maxn=1010;
int mp[maxn][maxn];
bool flag=false;
int n,m,xx,yy;
int dir[4][4]={{0,-1,0,1},{-1,0,1,0},{-1,-1,1,1},{1,-1,-1,1}};

bool check(int x,int y)
{
    if(x<1||x>n||y<1||y>n)
        return false;
    else
        return true;
}

void dfs(int x,int y,int dx,int dy,int cnt,int id)
{
    if(mp[x][y]!=id){
        xx=cnt;
        return;
    }
    cnt++;
    if(cnt==5){
        xx=cnt;
        return;
    }
    if(check(x+dx,y+dy))
        dfs(x+dx,y+dy,dx,dy,cnt,id);
    else
        xx=cnt;
}

int main()
{
    scanf("%d%d",&n,&m);
    int x,y,k=0;
    memset(mp,-1,sizeof(mp));
    for(int i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        mp[x][y]=i%2;
        if(flag&&k)
            continue;
        for(int j=0;j<4;j++){
            xx=0;
            dfs(x,y,dir[j][0],dir[j][1],0,i%2);
            yy=xx,xx=0;
            dfs(x,y,dir[j][2],dir[j][3],0,i%2);
//            printf("xx=%d   yy=%d\n",xx,yy);
            if(xx+yy-1>=5){
                //printf("xx+yy-1=%d\n",xx+yy-1);
                k=i;
                flag=true;
                break;
            }
        }
    }
    if(flag){
        if(k%2)
            printf("HtBest %d",k);
        else
            printf("WHZ %d",k);
    }
    else
        printf("UNK %d",m);
    return 0;
}