2018山东冬令营：网格II

问题 B: 网格II

时间限制: 1 Sec  内存限制: 128 MB
提交: 49  解决: 28
[提交][状态][讨论版]

题目描述

给出一个n*n的网格，有一些格子是障碍，再给出一对起点终点，求从起点到终点需要的最小步数，每次可以从一个格子走到上下左右4相邻的四个格子里。

输入

第一行一个整数n。
以下n行n个整数，描述整个网格，其中0表示没有障碍，1表示有障碍。
最后一行四个整数，Sx，Sy，Tx，Ty，描述起点和终点。

输出

输出最少步数。
如果永远走不到输出-1。

样例输入

3

0 1 0

 
0 1 0

0 0 0

1 1 1 3

样例输出

6

提示

对于30%的数据，n<=10
对于100%的测试数据，n<=1000。

【思路】

把坐标序列化后 就是 SPFA 了

【代码实现】

#include <iostream>
#include <bits/stdc++.h>
 
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
 
 
typedef long long ll;
const int MAXN=1e6+5;
const ll INF=0x3f3f3f3f;
int dir[5][2]={0,1,0,-1,1,0,-1,0};
int head[MAXN],dis[MAXN],vis[MAXN];
ll cot;
map<ll,ll>mp;
struct node{
    ll v,w,next;
}edge[MAXN<<1];
int maps[1005][1005];
ll a[1005][1005];
void init()
{
    cot=0;
    memset(head,-1,sizeof(head));
}
void add(ll u,ll v,int w)
{
    edge[++cot].v=v;
    edge[cot].w=w;
    edge[cot].next=head[u];
    head[u]=cot;
}
 
void SPFA(ll st)
{
    memset(vis,0,sizeof(vis));
    mem(dis,INF);
    queue <int> Q;
    vis[st]=0;
    dis[st]=0;
    Q.push(st);
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            int w=edge[i].w;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    vis[v]=1;
                    Q.push(v);
                }
            }
        }
    }
 
}
 
int main()
{
    init();
    int n;
    mem(maps,0);
    cin>>n;
    ll k=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>maps[i][j];
            a[i][j]=++k;
        }
    }
    int sx,sy,ex,ey;
    cin>>sx>>sy>>ex>>ey;
 
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(!maps[i][j])
                for(int k=0;k<4;k++)
                {
                    int px=i+dir[k][0];
                    int py=j+dir[k][1];
                    if(px>=1&&px<=n&&py>=1&&py<=n&&!maps[px][py])
                    {
                        add(a[i][j],a[px][py],1);
                    }
                }
        }
    }
    SPFA(a[sx][sy]);
    if(dis[a[ex][ey]]!=INF)
    {
        cout<<dis[a[ex][ey]]<<endl;
    }
    else
        cout<<-1<<endl;
    return 0;
}

123