문제

링크

풀이

#include <iostream>
#include <vector>
using namespace std;
 
int dx[4] = {0, 0, -1, 1};
int dy[4] = {-1, 1, 0, 0};
 
void solve(void) {
  int r, c, n; cin >> r >> c >> n; n--;
  vector<vector<int>> v(r, vector<int>(c));
  for (int i=0; i<r; i++) for (int j=0; j<c; j++) {
    char x; cin >> x;
    v[i][j] = x == 'O' ? 2 : 0;
  }
 
  while (n--) {
    for (int i=0; i<r; i++) for (int j=0; j<c; j++) {
      v[i][j] = v[i][j] ? v[i][j] - 1 : 3;
    }
    vector<vector<int>> tmp = v;
    for (int i=0; i<r; i++) for (int j=0; j<c; j++) {
      if (v[i][j]) continue;
      for (int k=0; k<4; k++) {
        int nx = i + dx[k], ny = j + dy[k];
        if (nx < 0 || nx >= r || ny < 0 || ny >= c) continue;
        tmp[nx][ny] = 0;
      }
    }
    v = tmp;
  }
 
  for (int i=0; i<r; i++) {
    for (int j=0; j<c; j++) {
      cout << (v[i][j] ? 'O' : '.');
    }
    cout << "\n";
  }
}
 
int main(void) {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
 
  solve();
  return 0;
}