和为零的子矩阵

* 给定一个整数矩阵,请找出一个子矩阵,使得其数字之和等于0.输出答案时,请返回左上数字和右下数字的坐标。
您在真实的面试中是否遇到过这个题? 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();
	  }
      
	}

}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值