結果

問題 No.124 門松列(3)
ユーザー yuppe19 😺yuppe19 😺
提出日時 2017-09-12 16:49:39
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 16 ms / 5,000 ms
コード長 1,669 bytes
コンパイル時間 931 ms
コンパイル使用メモリ 87,456 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-11-07 17:22:16
合計ジャッジ時間 1,971 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:19:18: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   19 |   int C, R; scanf("%d%d", &C, &R);
      |             ~~~~~^~~~~~~~~~~~~~~~
main.cpp:23:12: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   23 |       scanf("%d", &G[r][c]);
      |       ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <queue>
#include <tuple>
using namespace std;

                            // N  E  S  W
constexpr array<int, 4> dr = {-1, 0, 1, 0},
                        dc = { 0, 1, 0,-1};
constexpr int dr_size = dr.size();

bool is_kado(int a, int b, int c) {
  if(a == 0) { return true; }
  return a!=c && ((a>b&&b<c) || (a<b&&b>c));
}

int main(void) {
  constexpr int inf = 987654321;
  int C, R; scanf("%d%d", &C, &R);
  vector<vector<int>> G(R, vector<int>(C, 0));
  for(int r=0; r<R; ++r) {
    for(int c=0; c<C; ++c) {
      scanf("%d", &G[r][c]);
    }
  }
  vector<vector<vector<vector<int>>>> cost(R, vector<vector<vector<int>>>(C, vector<vector<int>>(10, vector<int>(10, inf)))); // cost[R][C][10][10]
  cost[0][0][0][G[0][0]] = 0;
  priority_queue<tuple<int, int, int, int, int>, vector<tuple<int, int, int, int, int>>, greater<tuple<int, int, int, int, int>>> pq;
  pq.emplace(0, 0, 0, 0, G[0][0]);
  while(!pq.empty()) {
    int _, r, c, b2, b1; tie(_, r, c, b2, b1) = pq.top(); pq.pop();
    for(int i=0; i<dr_size; ++i) {
      int nr = r + dr[i],
          nc = c + dc[i];
      if(!(0 <= nr && nr < R && 0 <= nc && nc < C)) { continue; }
      if(!is_kado(b2, b1, G[nr][nc])) { continue; }
      if(cost[nr][nc][b1][G[nr][nc]] > cost[r][c][b2][b1] + 1) {
        cost[nr][nc][b1][G[nr][nc]] = cost[r][c][b2][b1] + 1;
        pq.emplace(cost[nr][nc][b1][G[nr][nc]], nr, nc, b1, G[nr][nc]);
      }
    }
  }
  int res = inf;
  for(int b2=0; b2<10; ++b2) {
    for(int b1=0; b1<10; ++b1) {
      res = min(res, cost[R-1][C-1][b2][b1]);
    }
  }
  if(res == inf) { res = -1; }
  printf("%d\n", res);
  return 0;
}
0