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 Rust (judge verdict: Accepted)

use std::collections::VecDeque;

fn main(){
  let hwn: Vec<usize> = read_vec();
  let h = hwn[0];
  let w = hwn[1];
  let n = hwn[2] as i32;

  let mut mp: Vec<Vec<i32>> = vec![vec![-1; w+2]; h+2];
  
  for i in 1 .. h+1 {
    let ts: String = read();
    let tv: Vec<char> = ts.chars().collect();
    
    for j in 1 .. w+1 {
      mp[i][j] = match tv[j-1] {
        'S' => { 0 },
        'X' => { -1 },
        '.' => { 10 },
        c => { c as i32 - 48 },
      };
    }
  }

  let mut ans: i32 = 0;
  
  for i in 1 .. n+1 {
    ans += path(&mp, i);
  }
  
  println!("{}", ans);
}

fn read<T>() -> T
  where T: std::str::FromStr,
        T::Err: std::fmt::Debug
{
  let mut buf = String::new();
  std::io::stdin().read_line(&mut buf).expect("failed to read");
  buf.trim().parse().unwrap()
}

fn read_vec<T>() -> Vec<T>
  where T: std::str::FromStr,
        T::Err: std::fmt::Debug
{
  let mut buf = String::new();
  std::io::stdin().read_line(&mut buf).expect("failed to read");
  buf.split_whitespace().map(|e| e.parse().unwrap()).collect()
}

fn find_index(v: &Vec<Vec<i32>>, x: i32) -> (usize, usize) {
  let h = v.len();
  let w = v[0].len();
  
  for i in 1 .. h-1 {
    for j in 1 .. w-1 {
      if v[i][j] == x {
        return (i, j);
      }
    }
  }
  
  (0, 0)
}

fn path(v: &Vec<Vec<i32>>, x: i32) -> i32 {
  let h = v.len();
  let w = v[0].len();
  
  let (si, sj) = find_index(v, x-1);

  let mut pv: Vec<Vec<i32>> = vec![vec![-1;w];h];

  pv[si][sj] = 0;

  let mut que: VecDeque<(usize, usize)> = VecDeque::new();
  que.push_back((si, sj));

  loop {
    match que.pop_front() {
      Some((i, j)) => {
        let d: i32 = pv[i][j];
        
        if v[i][j] == x {
          return d;
        }

        if v[i+1][j] != -1 && pv[i+1][j] == -1 {
          pv[i+1][j] = d + 1;
          que.push_back((i + 1, j));
        }
        if v[i-1][j] != -1 && pv[i-1][j] == -1 {
          pv[i-1][j] = d + 1;
          que.push_back((i - 1, j));
        }
        if v[i][j+1] != -1 && pv[i][j+1] == -1 {
          pv[i][j+1] = d + 1;
          que.push_back((i, j + 1));
        }
        if v[i][j-1] != -1 && pv[i][j-1] == -1 {
          pv[i][j-1] = d + 1;
          que.push_back((i, j - 1));
        }
        
      },
      None => {return 0; },
    }
  }
}