提出 #486929

---

ソースコード 拡げる

/*
 * e.cc: E: 
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_H = 30;
const int MAX_W = 30;
const int DN = 8;
const int DBITS = 1 << DN;
const int DMASK = DBITS - 1;

const int INF = 1 << 30;

const int dxs[] = {1, 0, -1, 0}, dys[] = {0, -1, 0, 1};

/* typedef */

struct Stat {
  int x, y, bits, d;
  Stat() {}
  Stat(int _x, int _y, int _bits, int _d): x(_x), y(_y), bits(_bits), d(_d) {}
};

/* global variables */

int flds[MAX_H][MAX_W];
int dists[MAX_H][MAX_W][DBITS];
Stat prvs[MAX_H][MAX_W][DBITS];
int pds[DN];

/* subroutines */

/* main */

int main() {
  int h, w;
  cin >> h >> w;

  memset(flds, 0, sizeof(flds));
  int sx, sy, gx, gy;
  
  for (int y = 0; y < h; y++) {
    string line;
    cin >> line;
    for (int x = 0; x < w; x++) {
      if (line[x] == '#')
	flds[y][x] = -1;
      else if (line[x] == '^')
	sx = x, sy = y;
      else if (line[x] == '$')
	gx = x, gy = y;
      else if (line[x] >= 'a' && line[x] <= 'h')
	flds[y][x] = line[x] - 'a' + 1;
      else if (line[x] >= 'A' && line[x] <= 'H')
	flds[y][x] = line[x] - 'A' + 1 + DN;
    }
  }

  for (int y = 0; y < h; y++)
    for (int x = 0; x < w; x++)
      for (int bits = 0; bits < DBITS; bits++)
	dists[y][x][bits] = INF;
  dists[sy][sx][0] = 0;
  prvs[sy][sx][0] = Stat(sx, sy, 0, -1);
  
  queue<Stat> q;
  q.push(Stat(sx, sy, 0, 0));
  int gbits = -1;
  
  while (! q.empty()) {
    Stat u = q.front(); q.pop();
    if (dists[u.y][u.x][u.bits] != u.d) continue;
    if (u.x == gx && u.y == gy) {
      gbits = u.bits;
      break;
    }

    int vd = u.d + 1;
    
    for (int di = 0; di < 4; di++) {
      int vx = u.x + dxs[di], vy = u.y + dys[di];
      if (vx >= 0 && vx < w && vy >= 0 && vy < h && flds[vy][vx] >= 0) {
	int vbits = u.bits, vf = flds[vy][vx];
	if (vf >= 1 && vf <= DN)
	  vbits |= 1 << (vf - 1);
	else if (vf > DN && ! (vbits & (1 << (vf - DN - 1)))) continue;

	if (dists[vy][vx][vbits] > vd) {
	  dists[vy][vx][vbits] = vd;
	  q.push(Stat(vx, vy, vbits, vd));
	  prvs[vy][vx][vbits] = u;
	}
      }
    }
  }

  if (dists[gy][gx][gbits] >= INF) {
    cout << -1 << endl;
    return 0;
  }
  
  int mt = 0;
  memset(pds, 0, sizeof(pds));

  for (int y = gy, x = gx, bits = gbits; prvs[y][x][bits].d > 0;) {
    Stat u = prvs[y][x][bits];
    int uf = flds[u.y][u.x];
    if (uf >= 1 && uf <= DN) {
      int k = uf - 1;
      if (pds[k] > 0) {
	mt += pds[k] - u.d;
	for (int i = 0; i < DN; i++)
	  if (pds[i]) mt++;
	pds[k] = 0;
      }
    }
    else if (uf > DN) {
      int k = uf - DN - 1;
      if (pds[k] == 0) pds[k] = u.d;
    }

    x = u.x, y = u.y, bits = u.bits;
  }
  
  cout << (dists[gy][gx][gbits] + mt) << endl;

  return 0;
}

提出情報

ジャッジ結果