POJ1915,雙向寬度優先搜尋

Salvete發表於2018-08-30

第一道雙先寬度優先搜尋,過了很開心~

/*************************************************************************
    > File Name: main.cpp
    > Author:Eagles 
    > Mail:None 
    > Created Time: 2018年08月30日 星期四 10時07分17秒
    > Description:POJ1915,雙向寬度優先搜尋
 ************************************************************************/

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
#define N 301
int n;
int ans;

bool vis[2][N][N];//標記是否到達該點

int steps[2][N][N];//記錄到達每一個座標所需的步數

struct node
{
    int x,y;
    int step;
    node()
    {
        x=y=step=0;
    }
}star,en;

int dir[8][2]={{-2,1},{-2,-1},{-1,2},{-1,-2},{1,2},{1,-2},{2,1},{2,-1}};


bool check(int x, int y)
{
    return (0<=x&&x<n&&0<=y&&y<n);
}


bool  bfs()
{
    memset(vis,false,sizeof(vis));

    queue<node>Q[2];

    if (star.x==en.x&&star.y==en.y)
    {
        ans=0;
        return true;
    }

    Q[0].push(star);
    Q[1].push(en);

    vis[0][star.x][star.y]=true;
    vis[1][en.x][en.y]=true;
    steps[0][star.x][star.y]=0;
    steps[1][en.x][en.y]=0;

    node cur,nex;

    int inx=0;

    while (!Q[0].empty()||!Q[1].empty())
    {
        inx=(inx^1);//每一次換一個佇列進行擴充套件,亦可寫作inx=(inx+1)%2;

        int the_size=Q[inx].size();

        for (int i=0; i<the_size; i++)
        {
            cur=Q[inx].front();
            Q[inx].pop();

            for (int j=0; j<8; j++)
            {
                nex.x=cur.x+dir[j][0];
                nex.y=cur.y+dir[j][1];
                nex.step=cur.step+1;

                if (check(nex.x,nex.y))
                {
                    if (vis[inx^1][nex.x][nex.y])//如果已經相遇
                    {
                        ans=nex.step+steps[inx^1][nex.x][nex.y];
                        return true;
                    }

                    if (!vis[inx][nex.x][nex.y])
                    {

                        vis[inx][nex.x][nex.y]=true;
                        steps[inx][nex.x][nex.y]=nex.step;

                        Q[inx].push(nex);
                    }
                }
            }    


        }
    }
return false;
}


int main()
{

   // freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);

    int t;

    while(~scanf("%d",&t))
    {
        while(t--)
        {
            scanf("%d",&n);
            scanf("%d %d",&star.x,&star.y);
            scanf("%d %d",&en.x,&en.y);

            star.step=en.step=0;

            ans=-1;

            bfs();

            printf("%d\n",ans);
        }
    }

    return 0;
}

相關文章