Set Matrix Zeroes

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


题目的意思就是在一个矩阵中找到所有为零的点,然后把该点所在行与列都置为零。但是要求in place,就是原地置换。我理解的意思就是空间复杂度要求o(1)。


很水的一道题,只要遍历整个矩阵记录下为零的点就行了。但是要求空间复杂度o(1)的话可以使用第一行第一列来保留需要的置0的行数与列数。

先判断第一行与第一列是否需要置0,然后遍历矩阵。把需要置0的行数与列数保存在第一行与第一列中。然后把需要置0的行数与列数置0(这里要注意,不要修改第一行与第一列的值),最后通过之前的判断决定是否将第一行第一列置0。


void setZeroes(vector<vector<int> > &matrix) 
	{
		bool firstRow = false;
		bool firstColumn = false;
		for (auto i : matrix[0])
		{
			if (i == 0 )
			{
				firstRow = true;
				break;
			}
		}
		for (unsigned int i = 0; i < matrix.size(); i++)
		{
			if (matrix[i][0] == 0)
			{
				firstColumn = true;
				break;
			}
		}
		for (unsigned int i = 1; i < matrix.size(); i++)
			for (unsigned int j = 1; j < matrix[i].size(); j++)
			{
				if (matrix[i][j] == 0)
				{
					matrix[0][j] = 0;
					matrix[i][0] = 0;
				}
			}
		for (unsigned int i = 0; i < matrix[0].size(); i++)
			if (matrix[0][i] == 0)
				for (unsigned int j = 1; j < matrix.size(); j++)
					matrix[j][i] = 0;

		for (unsigned int i = 0; i < matrix.size(); i++)
			if (matrix[i][0] == 0)
				for (unsigned int j = 1; j < matrix[i].size(); j++)
					matrix[i][j] = 0;
		if (firstColumn)
			for (auto& i : matrix)
				i[0] = 0;
		if (firstRow)
			for (auto& i : matrix[0])
				i = 0;
	}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值