首页 > 其他 > 详细

Set Matrix Zeroes

时间:2014-03-12 22:56:49      阅读:548      评论:0      收藏:0      [点我收藏+]

题目原型:

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

click to show follow up.

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

基本思路:

空间复杂度为O(mn)和O(m + n) 的思路比较容易想,关键是第三种,空间复杂度要求是o(1),其实这是第二种思路的变形,即把第一行和第一列拿出来存储需要置0的行列的记录。

	//时间复杂度是O(1),借助了第一行和第一列,类似另外开辟空间来保存行和列的信息
	public void setZeroes(int[][] matrix)
	{
		if(matrix==null||matrix.length==0)
			return;
		boolean isZeroFstRow = false;
		boolean isZeroFstCol = false;
		//寻找第一行是否有0
		for(int i = 0;i<matrix[0].length;i++)
		{
			if(matrix[0][i]==0)
			{
				isZeroFstRow = true;
				break;
			}
		}
		//寻找第一列是否有0
		for(int i = 0;i<matrix.length;i++)
		{
			if(matrix[i][0]==0)
			{
				isZeroFstCol = true;
				break;
			}
		}
		//记录需要置0的行列
		for(int i = 1;i<matrix.length;i++)
		{
			for(int j = 1;j<matrix[0].length;j++)
			{
				if(matrix[i][j]==0)
				{
					matrix[i][0] = 0;
					matrix[0][j] = 0;
				}
			}
		}
		//置0
		for(int i = 1;i<matrix.length;i++)
		{
			for(int j = 1;j<matrix[0].length;j++)
			{
				if(matrix[i][0] == 0||matrix[0][j] == 0)
				{
					matrix[i][j] = 0;
				}
			}
		}
		//如果第一行存在0,则把第一行置0
		if(isZeroFstRow)
		{
			for(int i = 0;i<matrix[0].length;i++)
			{
				matrix[0][i] = 0;
			}
		}
		//如果第一列存在0,则把第一列置0
		if(isZeroFstCol)
		{
			for(int i = 0;i<matrix.length;i++)
			{
				matrix[i][0] = 0;
			}
		}
	}



Set Matrix Zeroes,布布扣,bubuko.com

Set Matrix Zeroes

原文:http://blog.csdn.net/cow__sky/article/details/21108863

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!