【题目描述】
Given an integer matrix, find a submatrix where the sum of numbers is zero. Your code should return the coordinate of the left-up and right-down number.
给定一个整数矩阵,请找出一个子矩阵,使得其数字之和等于0.输出答案时,请返回左上数字和右下数字的坐标。
【题目链接】
www.lintcode.com/en/problem/submatrix-sum/
【题目解析】
这道题和求数组中哪些元素和为0的解决方法一样,只是数组中求的是前i个元素和前j个元素和相等,则i-j元素和为0,而这里只是变成2维的而已。
sum[i][j]表示matrix[0][0]到matrix[i-1][j-1]所有元素的和。
建立sum矩阵,为n+1行,m+1列。将第0行和第0列都初始化为0。
遍历matrix,根据公式 sum[i][j] = matrix[i - 1][j - 1] + sum[i][j - 1] + sum[i - 1][j] -sum[i - 1][j - 1] 计算所有sum。
然后取两个row:l1, l2。用一个线k从左到右扫过l1和l2,每次都用diff=sum[l1][k]-sum[l2][k]来表示l1-l2和0-k这个矩形元素的sum。如果在同一个l1和l2中,有两条线(k1,k2)的diff相等,则表示l1-l2和k1-k2这个矩形中的元素和为0。
【参考答案】