思路:
由于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;
}
};