Leetcode 073 Set Matrix Zeroes

最新推荐文章于 2024-02-25 01:04:30 发布

JeraKrs

最新推荐文章于 2024-02-25 01:04:30 发布

阅读量153

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/keshuai19940722/article/details/80700317

版权

Leetcode 专栏收录该内容

151 篇文章 0 订阅

订阅专栏

题目连接：Leetcode 073 Set Matrix Zeroes

解题思路：原理是用一行和一列作为标记，然后用单独的两个变量记录它们是否被归零。我是用matrix[0][x]和matrix[y][0]作标记，遍历一遍数组，如果matrix[i][j]为零，那么标记matrix[0][j] = matrix[i][0] = 0。然后最后再遍历一遍数组，修改值，如果matrix[0][j] 或 matrix[i][0] = 0，则令matrix[i][j] = 0。

class Solution {
	public:
		void setZeroes(vector<vector<int>>& matrix) {
			if (matrix.size() == 0 || matrix[0].size() == 0) return ;

			int m = matrix.size(), n = matrix[0].size();
			bool row = false, col = false;
			for (int i = 0; i < n; i++) if (matrix[0][i] == 0) row = true;
			for (int i = 0; i < m; i++) if (matrix[i][0] == 0) col = true;

			for (int i = 1; i < m; i++) {
				for (int j = 1; j < n; j++) {
					if (matrix[i][j] == 0)
						matrix[i][0] = matrix[0][j] = 0;
				}
			}

			for (int i = 1; i < m; i++) {
				for (int j = 1; j < n; j++) {
					if (matrix[i][0] == 0 || matrix[0][j] == 0)
						matrix[i][j] = 0;
				}
			}

			if (row) for (int i = 0; i < n; i++) matrix[0][i] = 0;
			if (col) for (int i = 0; i < m; i++) matrix[i][0] = 0;
		}
};