Uriel's Corner

Research Associate @ Harvard University / Research Interests: Computer Vision, Biomedical Image Analysis, Machine Learning
posts - 0, comments - 50, trackbacks - 0, articles - 594
给出二维矩阵matrix,求其中满足矩阵和等于target的子矩阵有几个,O(n^4)暴力可过,计算矩阵的prefix sum再同时枚举所有可能


 1 #1074
 2 #Runtime: 9219 ms (Beats 7.69%)
 3 #Memory: 12.6 MB (Beats 100%)
 4 
 5 class Solution(object):
 6     def numSubmatrixSumTarget(self, matrix, target):
 7         """
 8         :type matrix: List[List[int]]
 9         :type target: int
10         :rtype: int
11         """
12         ans = 0
13         n, m = len(matrix), len(matrix[0])
14         pre_sum = [[0] * (m + 1) for _ in xrange(n + 1)]
15         for i in xrange(1, n + 1):
16             for j in xrange(1, m + 1):
17                 pre_sum[i][j] = pre_sum[i - 1][j] + pre_sum[i][j - 1] + matrix[i - 1][j - 1] - pre_sum[i - 1][j - 1]
18                 for x in xrange(0, i):
19                     for y in xrange(0, j):
20                         if target == pre_sum[i][j] + pre_sum[x][y] - pre_sum[x][j] - pre_sum[i][y]:
21                             ans += 1
22         return ans

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理