子矩阵数量

给出矩阵 matrix 和目标值 target,返回元素总和等于目标值的非空子矩阵的数量。

子矩阵 x1, y1, x2, y2 是满足 x1 <= x <= x2 且 y1 <= y <= y2 的所有单元 matrix[x][y] 的集合。

如果 (x1, y1, x2, y2) 和 (x1’, y1’, x2’, y2’) 两个子矩阵中部分坐标不同(如:x1 != x1’),那么这两个子矩阵也不同。

示例 1:

输入:matrix = [[0,1,0],[1,1,1],[0,1,0]], target = 0
输出:4
解释:四个只含 0 的 1x1 子矩阵。
示例 2:

输入:matrix = [[1,-1],[-1,1]], target = 0
输出:5
解释:两个 1x2 子矩阵,加上两个 2x1 子矩阵,再加上一个 2x2 子矩阵。
示例 3:

输入:matrix = [[904]], target = 0
输出:0

提示:

1 <= matrix.length <= 100
1 <= matrix[0].length <= 100
-1000 <= matrix[i] <= 1000
-10^8 <= target <= 10^8

题解:

class Solution {
public:
    void sums(int suml[105][105],vector<vector<int>> &matrix)
    {
        int rb = matrix.size(),cb=matrix[0].size();
        for(int i=0;i<rb;++i)
        {
            suml[i+1][1]=matrix[i][0];
            for(int j=1;j<cb;++j)
            {
                suml[i+1][j+1]=suml[i+1][j]+matrix[i][j];
            }
        }
        for(int i=2;i<=rb;++i)
        {
            for(int j=1;j<=cb;++j)
            {
                suml[i][j]+=suml[i-1][j];
            }
        }
    }
    int count(int r,int l,vector<vector<int>>& matrix,int target,int suml[105][105])
    {
        int rb = matrix.size(),cb=matrix[0].size();
        int res=0,ct=0;
        for(int i=r;i<rb;++i)
        {
            for(int j=l;j<cb;++j)
            {
                res=suml[i+1][j+1]+suml[i-r][j-l]-suml[i-r][j+1]-suml[i+1][j-l];
                if(res==target) ct++;
            }
        }
        return ct;
    }
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        int r,l;
        int rlen,llen;
        int ct=0;
        int array[105][105];
        for(int i=0;i<105;++i)
        for(int j=0;j<105;++j)
            array[i][j]=0;
        sums(array,matrix);
        rlen=matrix.size(),llen = matrix[0].size();
        for(int i=0;i<rlen;++i)
        {
            for(int j=0;j<llen;++j)
            {
                ct+=count(i,j,matrix,target,array);
            }
        }
        return ct;

    }
};

版权声明:本文为lh2018i原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。