元素和为目标值的子矩阵数量

Posted 2021-11-05 goto_1600

思路：
由于n和m都是100以内，我们只要控制复杂度在n^3以内就行了，我们可以枚举任意两列，然后枚举行，用个哈希表存一下这种类型的前缀和，算一下贡献就行了。

代码：

const int N=1010;
class Solution {
public:
    int  s[N][N];
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        memset(s,0,sizeof s);
        int n=matrix.size();
        int m=matrix[0].size();
        for(int i=1;i<=n;i++)
            for(int j=1;j <= m ; j ++)
            {
                s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + matrix[i-1][j-1];
            }
        int res=0;
        for(int i=1;i<=m;i++)
        {
            for(int j=i;j<=m;j++)
            {
                unordered_map<int,int> mp ;
                mp[0]++;
                for(int k=1;k<=n;k++)
                {
                    int tt = s[k][j] - s[k][i-1] ;
                   
                    res+=mp[-target+tt];
                     mp[tt]++;
                    //cout<<i<<" "<<j<<" "<<k<<" res="<<res<<" tt="<<tt<<endl;
                   
                }
            }
        }
        return res;
    }
};