LeetCode3212. 统计 X 和 Y 频数相等的子矩阵数量
我的解法
AC了
cpp
// 我的思路 :
// 条件要求包含grid[0][0],意思是这个矩阵一定是以grid[0][0]为左上角的,只用确定它的右上角即可确定唯一矩阵
// arr[i][j]的值=以grid[0][0]为左上角,以grid[i][j]为右上角的矩阵中'X'和'Y'的数量
// arr[i][j].first的值=以grid[0][0]为左上角,以grid[i][j]为右上角的矩阵中'X'的数量
// arr[i][j].second的值=以grid[0][0]为左上角,以grid[i][j]为右上角的矩阵中'Y'的数量
// 认真观察发现arr[i][j]的递推公式
// arr[i][j].first=(以grid[i-1][j]结束的矩阵的'X'数量)+(以grid[i][j-1]结束的矩阵的'X'数量)-(矩阵重复计算的'X'数量->arr[i-1][j-1].first)+(这个grid[i][j]是不是'X');
// arr[i][j].second=(以grid[i-1][j]结束的矩阵的'Y'数量)+(以grid[i][j-1]结束的矩阵的'Y'数量)-(矩阵重复计算的'Y'数量->arr[i-1][j-1].second)+(这个grid[i][j]是不是'Y');
// (注意边界问题)
class Solution {
public:
int numberOfSubmatrices(vector<vector<char>>& grid) {
int cnt=0;//统计符合条件的矩阵数量--->返回值
vector<vector<pair<int,int>>>arr(grid.size(),vector<pair<int,int>>(grid[0].size(),{0,0}) );//创建arr数组
for(int i=0;i<grid.size();++i){
for(int j=0;j<grid[0].size();++j){
arr[i][j].first=(i-1>=0?arr[i-1][j].first:0)+(j-1>=0?arr[i][j-1].first:0)-(i-1>=0&&j-1>=0?arr[i-1][j-1].first:0)+(grid[i][j]=='X'?1:0);
arr[i][j].second=(i-1>=0?arr[i-1][j].second:0)+(j-1>=0?arr[i][j-1].second:0)-(i-1>=0&&j-1>=0?arr[i-1][j-1].second:0)+(grid[i][j]=='Y'?1:0);
//三目运算符 哈哈哈
if(arr[i][j].first==arr[i][j].second&&arr[i][j].first>0)cnt++;//若满足('X数量=='Y'数量&&'X'数量>0)就计数++
}
}
return cnt;
}
};