POJ 3170--双重BFS

来源：互联网发布：网络发票查询编辑：程序博客网时间：2024/05/18 02:40

题意：

给一个 n 列 m行的矩阵，矩阵元素只有0,1,2,3,4，0表示可通过，1表示不可通过，矩阵中只有一个2和3，其他数字不限制，的问从2

开始到所有的4在折回到3的最短路是多少。

输入：

8 44 1 0 0 0 0 1 00 0 0 1 0 1 0 00 2 1 1 3 0 4 00 0 0 4 1 1 1 0

输出：

分析：

两个BFS，先从2到4，再从4到3.

代码：

#include <iostream>#include <queue>#include <cstdio>#include <cstring>#include <algorithm>#define inf 0x3f3f3f3fusing namespace std;const int maxn=1005;int a[maxn][maxn],vis[maxn][maxn],dist1[maxn][maxn],dist2[maxn][maxn];int n,m,ans;int dr[]= {0,0,1,-1};int dc[]= {1,-1,0,0};struct node{    int x,y;} s1,s2,t;void bfs(){    queue<node> q;    dist1[s1.x][s1.y]=0;    q.push(s1);    while(!q.empty())    {        node p=q.front();        q.pop();        int r=p.x;        int c=p.y;        for(int i=0; i<4; i++)        {            int rr=r+dr[i];            int cc=c+dc[i];            if(rr>=1&&rr<=m&&cc>=1&&cc<=n&&a[rr][cc]!=1&&dist1[rr][cc]==inf)            {                vis[rr][cc]=1;                t.x=rr;                t.y=cc;                dist1[rr][cc]=dist1[r][c]+1;                q.push(t);            }        }    }    while(!q.empty())        q.pop();    dist2[s2.x][s2.y]=0;    q.push(s2);    while(!q.empty())    {        node p=q.front();        q.pop();        int r=p.x;        int c=p.y;        for(int i=0; i<4; i++)        {            int rr=r+dr[i];            int cc=c+dc[i];            if(rr>=1&&rr<=m&&cc>=1&&cc<=n&&a[rr][cc]!=1&&dist2[rr][cc]==inf)            {                vis[rr][cc]=1;                t.x=rr;                t.y=cc;                dist2[rr][cc]=dist2[r][c]+1;                q.push(t);            }        }    }    for(int i=1; i<=m; i++)    {        for(int j=1; j<=n; j++)        {            if(a[i][j]==4&&dist1[i][j]!=inf&&dist2[i][j]!=inf)                ans=min(ans,dist1[i][j]+dist2[i][j]);        }    }}int main(){    scanf("%d%d",&n,&m);    for(int i=1; i<=m; i++)    {        for(int j=1; j<=n; j++)        {            scanf("%d",&a[i][j]);            if(a[i][j]==2)            {                s1.x=i;                s1.y=j;            }            if(a[i][j]==3)            {                s2.x=i;                s2.y=j;            }        }    }    memset(dist2,inf,sizeof(dist2));    memset(dist1,inf,sizeof(dist1));    ans=inf;    bfs();    printf("%d\n",ans);    return 0;}

0 0