[C++] 대회 장소 준비 (백준 9555번)

https://www.acmicpc.net/problem/9555

 

#include <iostream>
#include <vector>
using namespace std;

int dx[] = {-1, 0, 1, -1, 1, -1, 0, 1};
int dy[] = {1, 1, 1, 0, 0, -1, -1, -1};

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

    int t;
    cin >> t;
    while (t--) {
        int n, m;
        cin >> n >> m;
        bool chk[101] = {};

        vector<vector<int>> g(n, vector<int>(m));
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                cin >> g[i][j];
            }
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (chk[g[i][j]]) continue;
                for (int k = 0; k < 8; k++) {
                    int ni = i + dx[k];
                    int nj = j + dy[k];
                    if (ni >= 0 && ni < n && nj >= 0 && nj < m) {
                        if (g[ni][nj] == g[i][j]) {
                            chk[g[i][j]] = true;
                            break;
                        }
                    }
                }
            }
        }

        int cnt = 0;
        for (bool b: chk) if (b) cnt++;
        cout << cnt << '\n';
    }
}

문제에 명확히 나와있지는 않은 것 같으나, 팀 ID가 같으면 같은 학교 출신으로 생각하는 것 같다. 팀 ID는 1부터 100까지 가능하므로 bool 배열 chk[101]를 false로 초기화해뒀다. 그리고 자리 배치표에서 한칸 한칸 확인하며 이미 chk가 true이면 확인할 필요가 없고 chk가 false이면 주변 여덟 자리를 살펴보며 같은 ID가 있는지 확인한다. 만약 있다면 그 즉시 chk를 false로 바꾸고 break한다. (AC)