인프런 커뮤니티 질문&답변

안녕하세요 ㅎㅎ 주석 달았는데 참고 부탁드립니다.

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

int n, m;
int arr[101][101];
bool ok[101][101];
int dx[4] = {0, 0, 1, -1};
int dy[4] = {1, -1, 0, 0};
set<pair<int, int>> S;

void solve(int i, int j, set<pair<int, int>> &S) {
  queue<pair<int, int>> Q;
  Q.push({i, j});
  ok[i][j] = true;
  while (!Q.empty()) {
    auto cur = Q.front();
    Q.pop();
    for (int t = 0; t < 4; t++) {
      int nx = cur.first + dx[t];
      int ny = cur.second + dy[t];
      if (nx < 0 || nx >= n || ny < 0 || ny >= m || ok[nx][ny])
        continue;
      if (arr[nx][ny] == 1) {
        S.insert({nx, ny});
        continue;
      }
      ok[nx][ny] = true;
      Q.push({nx, ny});
    }
  }
}

int main(void) {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(NULL);

//GOOD
  cin >> n >> m;
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      cin >> arr[i][j];
    }
  }

  int cnt = 0;
  int last_cnt = 0;

  while (true) {
    S.clear(); 
    bool ok[101][101];
    // 이런식으로 초기화 해주는 것이 좋습니다.  
	memset(ok, sizeof(ok), 0);
    for (int i = 0; i < n; i++)
      for (int j = 0; j < m; j++)
        ok[i][j] = false;

// 왜 이렇게 하시죠? 이 문제는 가장자리가 모두 0이기 때문에 가장자리에서 한번만 solve를 작동하면 
// 됩니다. 
		// 서 -> 동
    for (int i = 0; i < n; i++) {
      for (int j = 0; j < m; j++) {
        if (arr[i][j] == 1 || ok[i][j]) {
          break;
        }
        solve(i, j, S);
      }
    }

		// 북 -> 서
    for (int j = 0; j < m; j++) {
      for (int i = 0; i < n; i++) {
        if (arr[i][j] == 1 || ok[i][j]) {
          break;
        }
        solve(i, j, S);
      }
    }

		// 남 -> 북
    for (int i = 0; i < n; i++) {
      for (int j = m - 1; j >= 0; j--) {
        if (arr[i][j] == 1 || ok[i][j]) {
          break;
        }
        solve(i, j, S);
      }
    }

		// 동 -> 서
    for (int j = 0; j < m; j++) {
      for (int i = n - 1; i >= 0; i--) {
        if (arr[i][j] == 1 || ok[i][j]) {
          break;
        }
        solve(i, j, S);
      }
    }
    if (S.size() == 0)
      break;
    cnt++;
    last_cnt = S.size();
    for (auto a : S)
      arr[a.first][a.second] = 0;
  }
  cout << cnt << "\n";
  cout << last_cnt;
  return 0;
}