POJ 3170--双重BFS

题意：

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

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

输入：

8 4
4 1 0 0 0 0 1 0
0 0 0 1 0 1 0 0
0 2 1 1 3 0 4 0
0 0 0 4 1 1 1 0

输出：

11

分析：

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

代码：

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define inf 0x3f3f3f3f
using 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;
}