行列ゼロ化 (C++ ソリューション)

トピック

行列の  要素が 0 の場合、その行と列のすべての要素を 0に設定します 。インプレース アルゴリズムを使用してください 。m x n

例 1:

入力:行列 = [[1,1,1],[1,0,1],[1,1,1]]
出力: [[1,0,1],[0,0,0],[1] ,0,1]]

例 2:

入力:行列 = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
出力: [[0,0,0,0],[0] ,4,5,0],[0,3,1,0]]

C++ コード

#include <iostream>
#include <vector>
using namespace std;

/*
* 定义矩阵置零函数
* 使用两个标记变量flag_row0,flag_col0分别记录第一行和第一列是否原本包含0
* 其他行列元素如果为0，改变元素所在第一行第一列为0
* 反过来使用第一行与第一列去更新其他行与列
* 使用两个标记变量更新第一行与第一列
*/
void setZeroes(vector<vector<int>>& matrix) {
	int m = matrix.size();
	int n = matrix[0].size();
	int flag_col0 = false, flag_row0 = false;
	for (int i = 0; i < m; ++i) {
		if (!matrix[i][0]) {
			flag_col0 = true;
		}
	}
	for (int j = 0; j < n; ++j) {
		if (!matrix[0][j]) {
			flag_row0 = true;
		}
	}
	for (int i = 1; i < m; ++i) {
		for (int j = 1; j < n; ++j) {
			if (!matrix[i][j]) {
				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] || !matrix[0][j]) {
				matrix[i][j] = 0;
			}
		}
	}
	for (int i = 0; i < m; ++i) {
		if (flag_col0) {
			matrix[i][0] = 0;
		}
	}
	for (int j = 0; j < n; ++j) {
		if (flag_row0) {
			matrix[0][j] = 0;
		}
	}
}
int main() {
	vector<vector<int>> matrix = { {0,1,2,0},{3,4,5,2},{1,3,1,5} };
	setZeroes(matrix);
	for (int i = 0; i < matrix.size(); ++i) {
		for (int j = 0; j < matrix[0].size(); ++j) {
			cout << matrix[i][j] << " ";
		}
		cout << endl;
	}
	return 0;
}

分析する

行列ゼロ化関数を定義し、2 つのフラグ変数 flag_row0 と flag_col0 を使用して、最初の行と列に元々 0 が含まれているかどうかを記録します。他の行と行の要素が 0 の場合は、要素の最初の行と最初の列を 0 に変更します。最初の行と最初の列を更新するために 2 つのタグ変数を使用して、最初の行と最初の列が他の行と列を更新します。