Problem:
チーズ (Cheese)

問題

今年も JOI 町のチーズ工場がチーズの生産を始め，ねずみが巣から顔を出した．JOI 町は東西南北に区画整理されていて，各区画は巣，チーズ工場，障害物，空き地のいずれかである．ねずみは巣から出発して全てのチーズ工場を訪れチーズを 1 個ずつ食べる．

この町には，N 個のチーズ工場があり，どの工場も１種類のチーズだけを生産している．チーズの硬さは工場によって異なっており，硬さ 1 から N までのチーズを生産するチーズ工場がちょうど 1 つずつある．

ねずみの最初の体力は 1 であり，チーズを 1 個食べるごとに体力が 1 増える．ただし，ねずみは自分の体力よりも硬いチーズを食べることはできない．

ねずみは，東西南北に隣り合う区画に 1 分で移動することができるが，障害物の区画には入ることができない．チーズ工場をチーズを食べずに通り過ぎることもできる．すべてのチーズを食べ終えるまでにかかる最短時間を求めるプログラムを書け．ただし，ねずみがチーズを食べるのにかかる時間は無視できる．

入力

入力は H+1 行ある．1 行目には 3 つの整数 H，W，N (1 ≤ H ≤ 1000，1 ≤ W ≤ 1000，1 ≤ N ≤ 9) がこの順に空白で区切られて書かれている．2 行目から H+1 行目までの各行には，'S'，'1', '2', ..., '9'，'X'，'.' からなる W 文字の文字列が書かれており，各々が各区画の状態を表している．北から i 番目，西から j 番目の区画を (i,j) と記述することにすると (1 ≤ i ≤ H, 1 ≤ j ≤ W)，第 i+1 行目の j 番目の文字は，区画 (i,j) が巣である場合は 'S' となり，障害物である場合は 'X' となり，空き地である場合は '.' となり，硬さ 1, 2, ..., 9 のチーズを生産する工場である場合はそれぞれ '1', '2', ..., '9' となる．入力には巣と硬さ 1, 2, ..., N のチーズを生産する工場がそれぞれ 1 つずつある．他のマスは障害物または空き地であることが保証されている．ねずみは全てのチーズを食べられることが保証されている．

出力

すべてのチーズを食べ終えるまでにかかる最短時間（分）を表す整数を 1 行で出力せよ．

入出力例

入力例 1

3 3 1
S..
...
..1

出力例 1

4

入力例 2

4 5 2
.X..1
....X
.XX.S
.2.X.

出力例 2

12

入力例 3

10 10 9
.X...X.S.X
6..5X..X1X
...XXXX..X
X..9X...X.
8.X2X..X3X
...XX.X4..
XX....7X..
X..X..XX..
X...X.XX..
..X.......

出力例 3

91

問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in C (judge verdict: Accepted)

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define DIR_CNT 4
int next_dir4[DIR_CNT][2] = {
  {-1, 0},
  {0, -1},
  {0, 1},
  {1, 0},
};

void get_int3(char *line, int size, int *a1, int *a2, int *a3) {
  if(!fgets(line, size, stdin)) return;
  sscanf(line, "%d %d %d", a1, a2, a3);
}

#define LEN_MAX 1000
#define BUF_SIZE LEN_MAX+3
#define ROW_MAX (LEN_MAX+2)
#define COL_MAX (LEN_MAX+2)


struct point {
  int x;
  int y;
  int dist;
};

struct point stack[ROW_MAX*COL_MAX];
int start = 0;;
int end = 0;

int is_empty(void) {
  return start == end;
}

void enqueue(struct point p) {
  stack[start++] = p;
}

struct point dequeue(void) {
  return stack[end++];
}

int main(void) {
  char line[BUF_SIZE];
  int row, col, cheeses;
  get_int3(line, BUF_SIZE, &row, &col, &cheeses);

  char map[ROW_MAX][COL_MAX];
  char checked[ROW_MAX][COL_MAX];
  int i, j;
  for(i = 0; i < ROW_MAX; i++) {
    memset(map[i], 'X', COL_MAX);
  }
  struct point ps[10];
  for(i = 1; i <= row; i++) {
    if(!fgets(line, BUF_SIZE, stdin)) return 0;
    for(j = 1; j <= col; j++) {
      map[i][j] = line[j-1];
      if(map[i][j] == 'S') {
        ps[0].x = i; ps[0].y = j; ps[0].dist = 0;
      }
      if(map[i][j] >= '1' && map[i][j] <= '9') {
        int idx = map[i][j] - '0';
        ps[idx].x = i; ps[idx].y = j;
      }
    }
  }

  for(i = 1; i <= cheeses; i++) {
    for(j = 0; j < ROW_MAX; j++) memset(checked[j], 0, COL_MAX);
    start = end = 0; // initialize queue
    enqueue(ps[i-1]);
    checked[ps[i-1].x][ps[i-1].y] = 1;
    while(!is_empty()) {
      struct point p = dequeue();
      if(map[p.x][p.y] == (i+'0')) {
        ps[i].dist = p.dist;
        break;
      }
      int ndist = p.dist + 1;
      for(j = 0; j < DIR_CNT; j++) {
        int next_x = p.x + next_dir4[j][0];
        int next_y = p.y + next_dir4[j][1];
        if(checked[next_x][next_y]) continue;
        if(map[next_x][next_y]=='X') continue;
        struct point np = {next_x, next_y, ndist};
        enqueue(np);
        checked[next_x][next_y] = 1;
      }
    }
  }
  printf("%d\n", ps[cheeses].dist);

  return 0;
}