Submission #1094176

#	Time	Username	Problem	Language	Result	Execution time	Memory
1094176		ramzialoulou	Tetris (COCI17_tetris)	C++17	80 / 80	0 ms	600 KiB

tetris

#include <bits/stdc++.h>

using namespace std;

#ifdef Ramzi
#include "debug.h"
#else
#define debug(...)
#endif

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int n, m;
  cin >> n >> m;
  vector<string> s(n);
  for (int i = 0; i < n; i++) {
    cin >> s[i];
  }
  auto safe = [&](int i, int j) -> int {
      return (i >= 0 && i < n && j >= 0 && j < m);
  };
  auto f1 = [&](int x, int y) -> int {
      vector<int> dx = {0, 1, 1};
      vector<int> dy = {1, 1, 0};
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          return 0;
        }
      }
      return 1;
  };
  auto f2 = [&](int x, int y) -> int {
      bool ret = true, ret2 = true;
      for (int i = 1; i < 4; i++) {
        if (!safe(x, y + i) || s[x][y + i] != s[x][y]) {
          ret = false;
        }
        if (!safe(x + i, y) || s[x + i][y] != s[x][y]) {
          ret2 = false;
        }
      }
      return (ret || ret2 ? 1 : 0);
  };
  auto f3 = [&](int x, int y) -> bool {
      vector<int> dx = {0, 1, 1};
      vector<int> dy = {1, 0, -1};
      bool ret = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret = false;
        }
      }
      dx = {1, 1, 2};
      dy = {0, 1, 1};
      bool ret2 = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret2 = false;
        }
      }
      return (ret || ret2 ? 1 : 0);
  };
  auto f4 = [&](int x, int y) {
      vector<int> dx = {0, 1, 1};
      vector<int> dy = {1, 1, 2};
      bool ret = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret = false;
        }
      }
      dx = {1, 1, 2};
      dy = {0, -1, -1};
      bool ret2 = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret2 = false;
        }
      }
      return (ret || ret2 ? 1 : 0);
  };
  auto f5 = [&](int x, int y) -> int {
      vector<int> dx = {1, 1, 1};
      vector<int> dy = {-1, 0, 1};
      bool ret = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret = false;
        }
      }
      dx = {0, 0, 1};
      dy = {1, 2, 1};
      bool ret2 = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret2 = false;
        }
      }
      dx = {1, 2, 1};
      dy = {0, 0, 1};
      bool ret3 = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret3 = false;
        }
      }
      dx = {1, 2, 1};
      dy = {0, 0, -1};
      bool ret4 = true;
      for (int i = 0; i < 3; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (!safe(nx, ny) || s[nx][ny] != s[x][y]) {
          ret4 = false;
        }
      }
      return (ret || ret2 || ret3 || ret4 ? 1 : 0);
  };
  vector<int> cnt(5);
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      if (s[i][j] != '.') {
        cnt[0] += f1(i, j);
        cnt[1] += f2(i, j);
        cnt[2] += f3(i, j);
        cnt[3] += f4(i, j);
        cnt[4] += f5(i, j);
      }
    }
  }
  for (int i = 0; i < 5; i++) {
    cout << cnt[i] << "\n";
  }
  return 0;
}

• Subtask 1: 80 / 80