编程练习——把含有0的行和列清零

请编写一个算法，若MxN矩阵中某个元素为0，则将其所在的行与列清零。

给定一个MxN的int[][]矩阵(C++中为vector<vector>)mat和矩阵的阶数n，请返回完成操作后的int[][]矩阵(C++中为vector<vector>)，保证n小于等于300，矩阵中的元素为int范围内。

思路：

1、先把含有0的位置保存

2、再根据保存的位置，进行清零

import java.util.*;
public class str5 {

	public int[][] clearzero(int[][] str,int n)
	{
		
		//首先找出含有0的行和列
		int h[] = new int[n];
		int l[] = new int[n];
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
			{
				if(str[i][j] ==0)
				{
					h[i]=1;
					l[j]=1;	
				}
				
			}
		
		
		//把含有0的行删除
		for(int i=0;i<n;i++)
		{
			if(h[i]==1)
			{
				for(int j=0;j<n;j++)
				{
					str[i][j]=0;
				}
				
			}
		}
		
		//把含有0的列删除
		for(int i=0;i<n;i++)
		{
			if(l[i]==1)
			{
			for(int j=0;j<n;j++)
			{
				str[j][i]=0;
			}
			}
		}
		
		return str;
	
	}
	public static void main(String[] args)
	{
		Scanner s = new Scanner(System.in);
		str5 str = new str5();
		int a[][] =  new int[3][3];
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			{
				a[i][j] = s.nextInt();
			}
		
		int b[][]=str.clearzero(a,3);
		
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			{
				System.out.println(b[i][j]);
			}
		
	}
	
	
}