题目描述
让我们来玩一个机器人游戏,游戏在一个长方形网格上进行,机器人最初被安放在长方形网格的左上角且面朝东,而游戏的目标就是使到达右下网格。
机器人可以执行以下5种操作:
“Straight”: 保持机器人当前的方向,并前进一格。
“Right”: 右转90度,并前进一格
“Back”: 转180度,并前进一格
“Left”: 左转90度,并前进一格
“Halt”: 停在原地,并结束游戏。
机器人可以经过同一个方块多次。如果你在游戏过程中,让机器人走出边界或者在到达目标之前执行了“Halt”指令,你都将失败。
现在你的任务是计算最小的代价,使机器人从起点到达终点。
输入
文件开始两个整数w,h,表示列和行数(2 ≤ h ≤ 30 ,2 ≤ w ≤ 30)
接下来共h行w列整数
最后4个整数c0 c1 c2 c3。
格式如下:
w h
s(1,1) … s(1,w)
s(2,1) … s(2,w)
…
s(h,1) … s(h,w)
c0 c1 c2 c3
矩阵中数字代表的含义如下:
0: “Straight”
1: “Right”
2: “Back”
3: “Left”
4: “Halt”
保证“Halt”命令将出现在目标位置,但也可能出现在矩阵中某个方格中。
最后4个数字c0, c1, c2, 和c3, 分别表示你使用”Straight”, “Right”, “Back”, and “Left” 命令的代价,其值在1~9之间。
输出
输出机器人从起点到终点的最小代价。
样例输入
8 3
0 0 0 0 0 0 0 1
2 3 0 1 4 0 0 1
3 3 0 0 0 0 0 4
9 9 1 9
样例输出
1
提示
30%数据w,h<10
100%数据w,h<=30
此题采用裂点方法,然后进行最短路即可。
裂点的裸题。
中间16个if语句可以通过循环减少代码复杂度。
Code
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int n,m,h,t,u,e,v,d1,d2,d3,d4,c0,c1,c2,c3,edgenum,ans;
int vet[100000],head[100000],len[100000],dis[100000],flag[100000],next[100000],q[100000];
int a[100][100];
int point(int x,int y,int z)
{
return ((x-1)*m+y-1)*4+z;
}
void add(int u,int v,int ok,int l)
{
vet[++edgenum]=v;
next[edgenum]=head[u];
len[edgenum]=(1-ok)*l;
head[u]=edgenum;
}
void spfa()
{
for(int i=1;i<=n*m*4;i++) dis[i]=1e9;
h=0;t=1;
q[1]=1;dis[1]=0;
while(h<t)
{
u=q[++h];
e=head[u];
while(e!=-1)
{
v=vet[e];
if(dis[u]+len[e]<dis[v])
{
dis[v]=dis[u]+len[e];
if(!flag[v])
{
q[++t]=v;
flag[v]=1;
}
}
e=next[e];
}
flag[u]=0;
}
}
int main()
{
scanf("%d%d",&m,&n);
for(int i=1;i<=n*m*4;i++) head[i]=-1;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
scanf("%d",&a[i][j]);
a[i][j]++;
}
scanf("%d%d%d%d",&c0,&c1,&c2,&c3);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
d1=point(i,j,1);
d2=point(i,j,2);
d3=point(i,j,3);
d4=point(i,j,4);
if(j+1<=m) add(d1,point(i,j+1,1),a[i][j]==1,c0);
if(i+1<=n) add(d2,point(i+1,j,2),a[i][j]==1,c0);
if(j-1>=1) add(d3,point(i,j-1,3),a[i][j]==1,c0);
if(i-1>=1) add(d4,point(i-1,j,4),a[i][j]==1,c0);
if(i+1<=n) add(d1,point(i+1,j,2),a[i][j]==2,c1);
if(j-1>=1) add(d2,point(i,j-1,3),a[i][j]==2,c1);
if(i-1>=1) add(d3,point(i-1,j,4),a[i][j]==2,c1);
if(j+1<=m) add(d4,point(i,j+1,1),a[i][j]==2,c1);
if(j-1>=1) add(d1,point(i,j-1,3),a[i][j]==3,c2);
if(i-1>=1) add(d2,point(i-1,j,4),a[i][j]==3,c2);
if(j+1<=m) add(d3,point(i,j+1,1),a[i][j]==3,c2);
if(i+1<=n) add(d4,point(i+1,j,2),a[i][j]==3,c2);
if(i-1>=1) add(d1,point(i-1,j,4),a[i][j]==4,c3);
if(j+1<=m) add(d2,point(i,j+1,1),a[i][j]==4,c3);
if(i+1<=n) add(d3,point(i+1,j,2),a[i][j]==4,c3);
if(j-1>=1) add(d4,point(i,j-1,3),a[i][j]==4,c3);
}
spfa();
ans=min(min(min(dis[point(n,m,1)],dis[point(n,m,2)]),dis[point(n,m,3)]),dis[point(n,m,4)]);
printf("%d\n",ans);
return 0;
}