提出 #33486377

---

ソースコード 拡げる

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

int main(void) {
  int H, W, N;
  cin >> H >> W >> N;
  vector<vector<int>> a(H, vector<int>(W));
  for (int i = 0; i < H; i++) {
    for (int j = 0; j < W; j++) {
      cin >> a[i][j];
      a[i][j]--;
    }
  }

  vector<int> c(N);
  for (int i = 0; i < N; i++) {
    cin >> c[i];
  }

  bool ok = true;
  vector<int> dx = {-1, 0, 1, 0};
  vector<int> dy = {0, 1, 0, -1};
  for (int i = 0; i < H; i++) {
    for (int j = 0; j < W; j++) {
      for (int k = 0; k < 4; k++) {
        int ni = i + dy[k], nj = j + dx[k];
        if (ni < 0 || ni >= H || nj < 0 || nj >= W) continue;
        if (a[ni][nj] != a[i][j] && c[a[ni][nj]] == c[a[i][j]]) {
          ok = false;
        }
      }
    }
  }
  cout << (ok ? "Yes" : "No") << endl;
  return 0;
}

提出情報

ジャッジ結果