【题解】【洛谷P1126】机器人搬重物

转载

Python全栈开发 2022-04-13 16:53:18 博主文章分类：fund

P1126 机器人搬重物

传送门

这道题本来没啥好说的，但细节实在比较多，被坑了好多次。

首先输入的是格子图，需要转化成点图，具体操作是
最坑的一个点在于，平时写宽搜的时候，遇到出边界或者不能访问的点时，都是直接进入下一层循环(continue)，但在这道题中，由于可以走1~3步，那么当路径上出现障碍时，则不能进行下一轮循环，需要break。

代码：

#include <bits/stdc++.h>
#define MAX 55
using namespace std;

int mod(int x){
  return (x+4)%4;
}

struct pt{
  int x, y, dir, step;
  pt(){}
  pt(int a, int b, int c, int d):x(a), y(b), dir(c), step(d){}
};

const int movx[] = {0,1,0,-1}, movy[] = {1,0,-1,0};
int a[MAX][MAX];
bool vis[MAX][MAX][5];
int n, m;
pt st, ed;

void bfs(){
  queue<pt> q;
  bool flag = false;
  st.step = 0;
  q.push(st);
  vis[st.x][st.y][st.dir] = true;
  while(!q.empty()){
    pt t = q.front();
    q.pop();
    if(t.x == ed.x && t.y == ed.y){
      cout << t.step << endl;
      flag = true;
      break;
    }
    for(int i = 1; i <= 3; i++){
      int u, v;
      u = t.x + i*movx[t.dir];
      v = t.y + i*movy[t.dir];
      if(u<=0 || u>=n || v<=0 || v>=m || a[u][v] == 1){
        break;
      }
      if(vis[u][v][t.dir]){
        continue;
      }
      vis[u][v][t.dir] = true;
      q.push(pt(u, v, t.dir, t.step+1));
    }
    if(!vis[t.x][t.y][mod(t.dir+1)]){
      vis[t.x][t.y][mod(t.dir+1)] = true;
      q.push(pt(t.x, t.y, mod(t.dir+1), t.step+1));
    }
    if(!vis[t.x][t.y][mod(t.dir-1)]){
      vis[t.x][t.y][mod(t.dir-1)] = true;
      q.push(pt(t.x, t.y, mod(t.dir-1), t.step+1));
    }
  }
  if(!flag){
    cout << -1 << endl;
  }
}

int main()
{
  cin >> n >> m;
  for(int i = 1; i <= n; i++){
    for(int j = 1; j <= m; j++){
      scanf("%d", &a[i][j]);
      if(a[i][j] == 1){
        a[i-1][j-1] = a[i-1][j] = a[i][j-1] = 1;
      }
    }
  }
  cin >> st.x >> st.y >> ed.x >> ed.y;
  char c;
  cin >> c;
  switch(c){
    case 'E':
      st.dir = 0; break;
    case 'S':
      st.dir = 1; break;
    case 'W':
      st.dir = 2; break;
    case 'N':
      st.dir = 3; break;
  }
  bfs();
  
  return 0;
}