P3930 SAC E#1 - 一道大水题 Knight

TLE，额 ，有空再写吧。

#include<queue>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
const int dx[]={,,-,-,,,-,-},dy[]={,-,,-,,-,,-};
const int N = ;
struct node{
    int x,y;
    int step,state;
};
queue<node>q;
int n,sx,sy;
char a[N][N];
int id[N][N],tot;
bool Vis[N][N][];
bool check( int  state,int x ,int  y)
{
    if(Vis[x][y][state])
        return ;
    //下
    for(int i=x+;i<=n;i++)
    if(a[i][y]!='.'&& !(state&(<<id[x][y])))
    {
        if(a[i][y]=='C'||a[i][y]=='Q')
        return ;
        break;
    }
    //上
    for(int i=x-;i<=;i--)
    if(a[i][y]!='.'&& !(state&(<<id[i][y])))
    {
        if(a[i][y]=='C'||a[i][y]=='Q')
        return ;
        break;
    }
    //左
    for(int j=y+;j<=n;j++)
    if(a[x][j]!='.'&&!(state&(<<id[x][j])))
    {
        if(a[x][j]=='C'||a[x][j]=='Q')
            return ;
        break;
    }
    //右
    for(int j=y-;j>=;j--)
    if(a[x][j]!='.'&&!(state&(<<id[x][j])))
    {
        if(a[x][j]=='C'||a[x][j]=='Q')
            return ;
        break;
    }
    //左上
    for(int i=x-,j=y-;i>=&&j>=;i--,j--)
    if(a[i][j]!='.'&&!(state&(<<id[i][j])))
    {
        if(a[i][j]=='B'||a[i][j]=='Q')
            return ;
        break;
    }
    //右上
    for(int i=x+,j=y-;i<=n&&j>=;i++,j--)
    if(a[i][j]!='.'&&!(state&(<<id[i][j])))
    {
        if(a[i][j]=='B'||a[i][j]=='Q')
            return ;
        break;
    }
    //左下
    for(int i=x-,j=y+;i>=&&j<=n;i--,j++)
    if(a[i][j]!='.'&&!(state&(<<id[i][j])))
    {
        if(a[i][j]=='B'||a[i][j]=='Q')
            return ;
        break;
    }
    //右下
    for(int i=x+,j=y+;i<=n&&j<=n;i++,j++)
    if(a[i][j]!='.'&&!(state&(<<id[i][j])))
    {
        if(a[i][j]=='B'||a[i][j]=='Q')
            return ;
        break;
    }
    int X,Y;
    for(int i=;i<=;i++)
    {
        X=x+dx[i];Y=y+dy[i];
        if(X>=&&X<=n&&Y>=&&Y<=n&&a[X][Y]=='K')
        return ;
    }
    if(x!=&&((y!=n&&a[x-][y+]=='P'&&!(state&(<<(id[x-][y+]))||(y!=&&a[x-][y-]=='P'&&!(state&(<<id[x-][y-])))))))
        return ;
    for(int i=x-;i<=x+;i++)
    for(int j=y-;j<=y+;j++)
    if(x>=&&x<=n&&y>=&&y<=n)
    {
        if(a[i][j]=='X')    return ;
    }
    return ;
}
void work()
{
    queue<node>q;tot=;
    for(int i=;i<=n;i++)
    for(int j=;j<=n;j++)
    {
        id[i][j]=-,a[i][j]=;
        for(int k=;k<=(<<);k++)
        Vis[i][j][k]=;
    }

    for(int i=;i<=n;i++)
    {
        scanf("%s",a[i]+);
        for(int j=;j<=n;j++)
        {
            if(a[i][j]=='O')
                sx=i,sy=j,a[i][j]='.';
            else
            if(a[i][j]=='C'||a[i][j]=='K'||a[i][j]=='Q'||a[i][j]=='X'||a[i][j]=='P')
                id[i][j]=tot++;
            else a[i][j]='.';
        }
    }
    if(!check(,sx,sy))
    {
        cout<<-;
        return;
    }
    q.push((node){sx,sy,,});
    node now;int x,y,state;
    while(!q.empty())
    {
        now=q.front();q.pop();
        state=now.state;
        for(int i=;i<;i++)
        {
            x=now.x+dx[i];y=now.y+dy[i];
            if(x<||x>n||y<||y>n)    continue;
            if(a[x][y]=='X')
            {
                printf("%d\n",now.step+);
                return;
            }
            if(a[x][y]!='.')
                state|= <<id[x][y];
            if(check(state,x,y))
                Vis[x][y][state]=,q.push((node){x,y,now.step+,state});
        }
    }
    cout<<-;
}
int main()
{
     while(scanf("%d",&n))
        work();
    return ;
}