美团2024届秋招笔试第一场编程真题
7.小美的好矩阵
小美定义一个矩阵是好矩阵,当且仅当该矩阵满足:
- 矩阵仅由’A’、‘B’、'C’三种字符组成。且三种字符都出现过。
- 矩阵相邻的字符都不相等。
现在给定一个n∗m的矩阵,小美想知道有多少个3*3的子矩阵是好矩阵,你能帮帮她吗?
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 256M,其他语言512M
输入描述:
第一行输入两个整数n,m,代表矩阵的行数和列数。
接下来的n行,每行输入一个仅包含大写字母的长度为m的字符串。
1\leq n,m \leq 1000
输出描述:
输出一个整数表示答案。
示例1
输入例子:
4 4
DABC
ABAB
BABA
BBAB
输出例子:
1
例子说明:
有4个3*3的子矩阵。
左上角的子矩阵出现了’D’,因此不合法。
右上角的是好矩阵。
左下角的存在两个相邻的字母相同,因此不合法。
右下角的子矩阵里没有’C’,因此不合法。
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, m;
cin >> n >> m;
vector<vector<char>> a(n, vector<char>(m));
vector<vector<int>> f(n, vector<int>(m, 0));
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
cin >> a[i][j];
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if ((j + 1 < m && a[i][j] == a[i][j + 1])) {
f[i]