Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Solution {
- public:
- int numSubmatrixSumTarget(vector<vector<int>>& A, int target) {
- int res = 0, m = A.size(), n = A[0].size();
- for (int i = 0; i < m; i++)
- for (int j = 1; j < n; j++)
- A[i][j] += A[i][j - 1];
- unordered_map<int, int> counter;
- for (int i = 0; i < n; i++) {
- for (int j = i; j < n; j++) {
- counter = {{0,1}}; // Important, don't miss out this line.
- int cur = 0;
- for (int k = 0; k < m; k++) {
- cur += A[k][j] - (i > 0 ? A[k][i - 1] : 0);
- res += counter.find(cur - target) != counter.end() ? counter[cur - target] : 0;
- counter[cur]++;
- }
- }
- }
- return res;
- }
- };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement