[C++/28256] 초콜릿 보관함

---

⛔ 단순히 기록용 입니다... 어떻게 풀었는가 생각도 다시 해보고 그러니까 아마도 도움은 안되실 것 같습니다.

---

범위도 3x3으로 정해져 있고, 가운데를 제외한 나머지 초콜릿이 그룹을 어떻게 이루어져 있는지 파악하는 문제입니다. 구현할게 은근히 많지만 그래도 기본적인 구현공부를 했던 사람이라면? 충분히 풀어낼 수 있는 문제입니다.

하지만, 나는 여기서 한 번 틀리고 마는데 이유는 무엇이냐면? 화면에 표시된 숫자의 개수인 n이다. n은 0 부터 4 까지이며 그리고 다음 문제는 이것이다. 정답을 틀리게 줄 수 있는 것이다. 개수가 같아도 틀릴 수 있고, 등등 여러가지 조건이 붙어 있다. 아래는 정답 코드이다.

#include<iostream>
#include<queue>
#include<vector>
#include<algorithm>
#include<cstring> 

#define MAX 4
using namespace std;

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

int a,testCase,board[MAX][MAX],visited[MAX][MAX];
vector<int> answer,example;
pair<int,int> center = {1,1};

void init() {
   
   cin >> testCase;
   
}

void printVisited() {
   
   cout << "print visited!" << "\n";
   
   for(int i = 0; i < 3; i++){
      for(int j = 0; j < 3; j++){
         cout << visited[i][j] << " ";
      }
      cout << "\n";
   }
   
   
}

int isRange(int y, int x){
   return 0 <= y && y < 3 && 0 <= x && x < 3;
}

void bfs(int y, int x){

   queue<pair<int,int> > q;
   int cnt = 1;
   q.push({y,x});
   visited[y][x] = 1;
   
   while(!q.empty()){
      
      int y = q.front().first;
      int x = q.front().second;
      q.pop();
      
      for(int i = 0; i < 4; i++){
         
         int ny = y + dy[i];
         int nx = x + dx[i];
         
         if(!isRange(ny,nx)) continue;
         if(visited[ny][nx]) continue;
         if(!board[ny][nx]) continue;
         if(ny == center.first && nx == center.second) continue;
         
         cnt++;
         visited[ny][nx] = 1;
         q.push({ny,nx});
         
      }
      
   }

   answer.push_back(cnt);

}

void solve() {
   
   memset(visited,0,sizeof(visited));
   
   for(int i = 0; i < 3; i++){
      for(int j = 0; j < 3; j++){
         if(!visited[i][j] && board[i][j] == 1){
            bfs(i,j);
         }
      }
   }

}

int main() {
   ios::sync_with_stdio(false);
	cin.tie(NULL);
   init();
   
   while(testCase--){
      
      example.clear();
      answer.clear();
      
      for(int i = 0; i < 3; i++){
         for(int j = 0; j < 3; j++){
            char c; cin >> c;
            
            if(c == 'O'){
               board[i][j] = 1;
            } else if(c == '-'){
               board[i][j] = 2;
            } else if(c == 'X'){
               board[i][j] = 0;
            }
         }
      }
      
      cin >> a;
      
      for(int i = 0; i < a; i++){
         int number; cin >> number;
         example.push_back(number);
      }
      
      solve();
      sort(answer.begin(), answer.end());
      
      int chk = true;
      if(example.size() == 0 && answer.size() >= 1) chk = false;
      if(!example.size() && !answer.size()) chk = true;
      
      for(int i = 0; i < example.size(); i++){
         
         if(example.size() != answer.size()) {
            chk = false;
            break;
         }
         
         if(example[i] != answer[i]){
            chk = false;
            break;
         }
      }
      
      cout << chk << "\n";
       
   }

}