#include #include #include using namespace std; vector>> generate_tetrominos() { vector>> tetrominos = { {{0, 0}, {0, 1}, {0, 2}, {0, 3}}, // I {{0, 0}, {1, 0}, {2, 0}, {3, 0}}, // I rotated 90 {{0, 0}, {1, 0}, {0, 1}, {1, 1}}, // O {{0, 0}, {0, 1}, {1, 0}, {1, 1}}, // O rotated 90 (same as O) {{0, 0}, {0, 1}, {0, 2}, {1, 0}}, // L {{0, 0}, {1, 0}, {2, 0}, {0, 1}}, // L rotated 90 {{0, 2}, {1, 0}, {1, 1}, {1, 2}}, // L rotated 180 {{0, 0}, {0, 1}, {1, 1}, {2, 1}}, // L rotated 270 {{0, 0}, {0, 1}, {0, 2}, {1, 2}}, // J {{0, 1}, {1, 1}, {2, 0}, {2, 1}}, // J rotated 90 {{0, 0}, {1, 0}, {1, 1}, {1, 2}}, // J rotated 180 {{0, 0}, {0, 1}, {1, 0}, {2, 0}}, // J rotated 270 {{0, 0}, {0, 1}, {1, 1}, {1, 2}}, // Z {{0, 1}, {1, 0}, {1, 1}, {2, 0}}, // Z rotated 90 {{0, 0}, {0, 1}, {-1, 1}, {-1, 2}}, // S {{0, 0}, {1, 0}, {1, 1}, {2, 1}}, // S rotated 90 {{0, 0}, {0, 1}, {0, 2}, {1, 1}}, // T {{0, 0}, {1, 0}, {1, 1}, {2, 0}}, // T rotated 90 {{0, 1}, {1, 0}, {1, 1}, {1, 2}}, // T rotated 180 {{0, 1}, {1, 0}, {1, 1}, {2, 1}} // T rotated 270 }; return tetrominos; } int max_tetromino_sum(const vector>& matrix, int n, int m) { auto tetrominos = generate_tetrominos(); int max_sum = 0; for (int i = 0; i < n; ++i) { for (int j = 0; j < m; ++j) { for (const auto& tetromino : tetrominos) { int current_sum = 0; bool valid = true; for (const auto& [dx, dy] : tetromino) { int x = i + dx, y = j + dy; if (x >= 0 && x < n && y >= 0 && y < m) { current_sum += matrix[x][y]; } else { valid = false; break; } } if (valid) { max_sum = max(max_sum, current_sum); } } } } return max_sum; } int main() { int n, m; cin >> n >> m; vector> matrix(n, vector(m)); for (int i = 0; i < n; ++i) { for (int j = 0; j < m; ++j) { cin >> matrix[i][j]; } } cout << max_tetromino_sum(matrix, n, m) << endl; return 0; }

제출 #12640

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력: