* 给定一个整数矩阵,请找出一个子矩阵,使得其数字之和等于0.输出答案时,请返回左上数字和右下数字的坐标。
您在真实的面试中是否遇到过这个题? Yes
样例
给定矩阵
[
[1 ,5 ,7],
[3 ,7 ,-8],
[4 ,-8 ,9],
]
您在真实的面试中是否遇到过这个题? Yes
样例
给定矩阵
[
[1 ,5 ,7],
[3 ,7 ,-8],
[4 ,-8 ,9],
]
返回 [(1,1), (2,2)]
import java.util.HashMap;
import java.util.Scanner;
/**
* 给定一个整数矩阵,请找出一个子矩阵,使得其数字之和等于0.输出答案时,请返回左上数字和右下数字的坐标。
您在真实的面试中是否遇到过这个题? Yes
样例
给定矩阵
[
[1 ,5 ,7],
[3 ,7 ,-8],
[4 ,-8 ,9],
]
返回 [(1,1), (2,2)]
*
* @author Dell
*
*/
public class Test405 {
//可怕的六重循环,好理解,但是时间复杂度太高。
public static int[][] submatrixSum(int[][] matrix)
{
int m=matrix.length;
if(m==0)
return null;
int n=matrix[0].length;
int[][] result=new int[2][2];
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
int start_x=i;
int start_y=j;
for(int k=i;k<m;k++)
{
for(int l=j+1;l<n;l++)
{
int end_x=k;
int end_y=l;
int sum=0;
for(int c=start_x;c<=end_x;c++)
{
for(int v=start_y;v<=end_y;v++)
{
sum=sum+matrix[c][v];
}
}
if(sum==0)
{
result[0][0]=start_x;
result[0][1]=start_y;
result[1][0]=end_x;
result[1][1]=end_y;
return result;
}
}
}
}
}
return result;
}
public static int[][] submatrixSum1(int[][] matrix)
{
int[][] result=new int[2][2];
int m=matrix.length;
if(m==0)
return result;
int n=matrix[0].length;
if(n==0)
return result;
int[][] sum=new int[m+1][n+1];
for(int i=0;i<=m;i++)
sum[i][0]=0;
for(int j=0;j<=n;j++)
sum[0][j]=0;
for(int i=1;i<=m;i++)
{
for(int j=1;j<=n;j++)
{
sum[i][j]=matrix[i-1][j-1]+sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];
}
}
for(int l=0;l<m;l++)
{
for(int h=l+1;h<=m;h++)
{
HashMap<Integer,Integer> map=new HashMap<>();
for(int j=0;j<=n;j++)
{
int diff=sum[h][j]-sum[l][j];
if(map.containsKey(diff))
{
int k=map.get(diff);
result[0][0]=l;
result[0][1]=k;
result[1][0]=h-1;
result[1][1]=j-1;
return result;
}
else
{
map.put(diff, j);
}
}
}
}
return result;
}
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int m=sc.nextInt();
int n=sc.nextInt();
int[][] a=new int[m][n];
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
a[i][j]=sc.nextInt();
}
}
int[][] result=submatrixSum1(a);
for(int i=0;i<result.length;i++)
{
for(int j=0;j<result[0].length;j++)
{
System.out.print(result[i][j]+" ");
}
System.out.println();
}
}
}