问题 B: 网格II
时间限制: 1 Sec 内存限制: 128 MB提交: 49 解决: 28
[ 提交][ 状态][ 讨论版]
题目描述
给出一个n*n的网格,有一些格子是障碍,再给出一对起点终点,求从起点到终点需要的最小步数,每次可以从一个格子走到上下左右4相邻的四个格子里。
输入
第一行一个整数n。
以下n行n个整数,描述整个网格,其中0表示没有障碍,1表示有障碍。
最后一行四个整数,Sx,Sy,Tx,Ty,描述起点和终点。
以下n行n个整数,描述整个网格,其中0表示没有障碍,1表示有障碍。
最后一行四个整数,Sx,Sy,Tx,Ty,描述起点和终点。
输出
输出最少步数。
如果永远走不到输出-1。
如果永远走不到输出-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