3212力扣:统计X和Y频数相等的子矩阵数量

cpp 复制代码
class Solution {
public:
    int numberOfSubmatrices(vector<vector<char>>& grid) {
        int n=grid.size();
        int m=grid[0].size();
        vector<vector<int>> prex(n+1,vector<int>(m+1,0));
        vector<vector<int>> prey(n+1,vector<int>(m+1,0));
        for(int i=1;i<n+1;i++){
            for(int j=1;j<m+1;j++){
                if(grid[i-1][j-1]=='X'){
                    prex[i][j]=prex[i-1][j]+prex[i][j-1]-prex[i-1][j-1]+1;}
                else{
                    prex[i][j]=prex[i-1][j]+prex[i][j-1]-prex[i-1][j-1]+0;
                }
                if(grid[i-1][j-1]=='Y'){
                    prey[i][j]=prey[i-1][j]+prey[i][j-1]-prey[i-1][j-1]+1;
                }else{
                    prey[i][j]=prey[i-1][j]+prey[i][j-1]-prey[i-1][j-1]+0;
                }
            }
        }
        int res=0;
        for(int i=1;i<n+1;i++){
            for(int j=1;j<m+1;j++){
                int countx=prex[i][j]-prex[0][j]-prex[i][0]+prex[0][0];
                int county=prey[i][j]-prey[0][j]-prey[i][0]+prey[0][0];
                if(countx==county && countx>0){res++;}
            }
        }
        return res;
    }
};

二维前缀和 前缀和数组定义时扩宽一层

相关推荐
剑指offer.26 分钟前
Linux多任务-线程篇
java·jvm·算法
-dzk-33 分钟前
【矩阵】LC 54.螺旋矩阵
线性代数·矩阵
nike0good42 分钟前
CF 126B(Password-z algorithm/exkmp)
开发语言·c++·算法
无定义_1 小时前
Bellman-Ford——贝尔曼福特算法
数据库·算法
HanhahnaH2 小时前
各数据结构操作的时间复杂度汇总
数据结构·算法
Yzzz-F2 小时前
CF2023D
c++·算法·dp
小小帅呀2 小时前
学习 VLA 第6天:SWIN VIT算法原理以及复现
学习·算法
猎头南楼3 小时前
杭州算法工程师,偏大模型应用落地
算法
间歇性努力持续性发呆的野生快乐选手3 小时前
二分模板(整型,浮点型)
数据结构·c++·算法