Advertisement
Josif_tepe

Untitled

Oct 28th, 2024
45
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.91 KB | None | 0 0
  1. #include <iostream>
  2. #include <queue>
  3. #include <vector>
  4. using namespace std;
  5.  
  6. int main() {
  7.     int n, m;
  8.     cin >> n >> m;
  9.    
  10.     char mat[n][m];
  11.     for(int i = 0; i < n; i++) {
  12.         for(int j = 0; j < m; j++) {
  13.             cin >> mat[i][j];
  14.         }
  15.     }
  16.     int di[] = {-1, 1, 0, 0};
  17.     int dj[] = {0, 0, -1, 1};
  18.    
  19.     int res_ovci = 0, res_volci = 0;
  20.     vector<vector<bool>> visited(n, vector<bool>(m, false));
  21.     for(int i = 0 ; i < n; i++) {
  22.         for(int j = 0; j < m; j++) {
  23.             if(mat[i][j] != '#' and !visited[i][j]) {
  24.                 queue<int> q;
  25.                 q.push(i);
  26.                 q.push(j);
  27.                 int ovci = 0, volci = 0;
  28.                 visited[i][j] = true;
  29.                
  30.                 while(!q.empty()) {
  31.                     int ci = q.front();
  32.                     q.pop();
  33.                     int cj = q.front();
  34.                     q.pop();
  35.                    
  36.                     if(mat[ci][cj] == 'o') {
  37.                         ovci++;
  38.                     }
  39.                     if(mat[ci][cj] == 'v') {
  40.                         volci++;
  41.                     }
  42.                     for(int k = 0; k < 4; k++) {
  43.                         int ti = ci + di[k];
  44.                         int tj = cj + dj[k];
  45.                        
  46.                         if(ti >= 0 and ti < n and tj >= 0 and tj < m and !visited[ti][tj] and mat[ti][tj] != '#') {
  47.                             q.push(ti);
  48.                             q.push(tj);
  49.                             visited[ti][tj] = true;
  50.                         }
  51.                     }
  52.                    
  53.                 }
  54.                 if(ovci > volci) {
  55.                     res_ovci += ovci;
  56.                 }
  57.                 else {
  58.                     res_volci += volci;
  59.                 }
  60.             }
  61.         }
  62.     }
  63.     cout << res_ovci << " " << res_volci << endl;
  64.     return 0;
  65. }
  66.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement