TakahiroNakamori

TakahiroNakamori

中森崇博(ナカモリタカヒロ)

12 /
08
2021
 

埋め立て

問題

B – 埋め立て
プログラミング初級者から上級者まで楽しめる、競技プログラミングコンテストサイト「AtCoder」。オンラインで毎週開催プログラミングコンテストを開催しています。競技…

解法

  1. oの数を数えておく。
  2. つながっているoの数を数える。
  3. xについて、1つずつoだと仮定して、つながっているoの数を数える。
  4. つながっているoの数はBFSで数える。

コード例

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using P = pair<int, int>;
#define rep(i, n) for(int i = 0; i < (n); ++i)
// using mint = modint998244353;
// using mint = modint1000000007;
// const int mod = 1000000007;
// const ll INF = 1LL << 60;
const int INF = 1001001001;

// CADDi 2018
// https://atcoder.jp/contests/caddi2018

int H = 10;
int W = 10;

vector<string> A;

deque<P> que;

vector<int> dy = {-1, 0, 1, 0};
vector<int> dx = {0, -1, 0, 1};

int f() {
  int result = 0;
  vector<vector<int>> check(10, vector<int>(10, INF));

  while(!que.empty()) {
    int h = que.front().first;
    int w = que.front().second;
    que.pop_front();

    for(int i = 0; i < 4; ++i) {
      int nh = h + dy[i];
      int nw = w + dx[i];
      if(nh < 0 || H <= nh || nw < 0 || W <= nw) {
        continue;
      }
      if(A[nh][nw] == 'x') {
        continue;
      }
      if(check[nh][nw] != INF) {
        continue;
      }
      result++;
      check[nh][nw] = 1;
      que.emplace_back(nh, nw);
    }
  }

  return result;
}

int main() {
  int sum = 0;
  vector<P> p;
  P start;
  A.resize(10);

  rep(i, H) {
    cin >> A[i];
    rep(j, W) {
      if(A[i][j] == 'o') {
        sum++;
        if(sum == 1) {
          start = make_pair(i, j);
        }
      } else {
        p.push_back(make_pair(i, j));
      }
    }
  }

  que.emplace_back(start.first, start.second);
  int v = f();
  if(v == sum) {
    cout << "YES" << endl;
    return 0;
  }

  int n = p.size();

  rep(i, n) {
    int tH = p[i].first;
    int tW = p[i].second;
    A[tH][tW] = 'o';
    que.emplace_back(start.first, start.second);
    int v = f();
    A[tH][tW] = 'x';
    if(v == sum + 1) {
      cout << "YES" << endl;
      return 0;
    }
  }

  cout << "NO" << endl;
  return 0;
}