oj.uz

답안 #319716

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
319716 2020-11-06T09:03:59 Z gustason Tetris (COCI17_tetris) C++14
80 / 80
 1 ms 708 KB 
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int r, s;
char grid[11][11];

vector<string> shape1{"11",
                      "11"};

vector<string> shape2{"1111"};

vector<string> shape3{"011",
                      "110"};

vector<string> shape4{"110",
                      "011"};

vector<string> shape5{"010",
                      "111"};

const vector<vector<string>> shapes = {shape1, shape2, shape3, shape4, shape5};
const vector<int> rot = {1, 2, 2, 2, 4};

vector<string> rotate(vector<string> mat) {
  int r = mat.size(), s = mat[0].size();
  vector<string> ret;

  ret.resize(s);
  for (int i = 0; i < ret.size(); ++i)
    ret[i].resize(r);

  for (int i = 0; i < r; ++i)
    for (int j = 0; j < s; ++j)
      ret[j][r - 1 - i] = mat[i][j];

  return ret;
}

int count(vector<string> shape) {
  int rr = shape.size(), ss = shape[0].size();
  int ret = 0;
  for (int i = 0; i < r; ++i) {
    for (int j = 0; j < s; ++j) {
      if (i + rr > r || j + ss > s) continue;
      set <char> S;
      for (int _r = i; _r < i + rr; ++_r)
        for (int _s = j; _s < j + ss; ++_s)
          if (shape[_r - i][_s - j] == '1')
            S.insert(grid[_r][_s]);
      ret += (S.size() == 1 && *S.begin() != '.');
    }
  }
  return ret;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> r >> s;
    for(int i = 0; i < r; i++) {
        for(int j = 0; j < s; j++) {
            cin >> grid[i][j];
        }
    }

    for(int i = 0; i <= 4; i++) {
        auto shape = shapes[i];
        int cnt = 0;
        for(int j = 0; j < rot[i]; j++) {
            cnt += count(shape);
            shape = rotate(shape);
        }
        cout << cnt << "\n";
    }
    return 0;
}
//~ check for overflows

Compilation message

tetris.cpp: In function 'std::vector<std::__cxx11::basic_string<char> > rotate(std::vector<std::__cxx11::basic_string<char> >)':
tetris.cpp:29:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::__cxx11::basic_string<char> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |   for (int i = 0; i < ret.size(); ++i)
      |                   ~~^~~~~~~~~~~~

Subtask #1
80.0 / 80.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 396 KB Output is correct
2 Correct 1 ms 452 KB Output is correct
3 Correct 1 ms 364 KB Output is correct
4 Correct 1 ms 708 KB Output is correct
5 Correct 1 ms 364 KB Output is correct
6 Correct 1 ms 492 KB Output is correct
7 Correct 1 ms 364 KB Output is correct
8 Correct 0 ms 404 KB Output is correct
9 Correct 1 ms 364 KB Output is correct
10 Correct 1 ms 364 KB Output is correct