每日刷题20210225—867. 转置矩阵 73. 矩阵置零

867. 转置矩阵

给你一个二维整数数组 matrix， 返回 matrix 的 转置矩阵 。
矩阵的 转置 是指将矩阵的主对角线翻转，交换矩阵的行索引与列索引。
示例 1：
输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[[1,4,7],[2,5,8],[3,6,9]]
示例 2：
输入：matrix = [[1,2,3],[4,5,6]]
输出：[[1,4],[2,5],[3,6]]

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& matrix) {
		int m = matrix.size();
		int n = matrix[0].size();
		 
		vector<vector<int>> rlt(n, vector<int>(m));
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				rlt[i][j] = matrix[j][i];
			}
		}
		return rlt;
    }
};

73. 矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 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]
]

进阶:
一个直接的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗？

class Solution {
public:
    	void setZeroes(vector<vector<int>>& matrix) {
        int n = matrix.size();
		int m = matrix[0].size();
		int wSetflag = 0, hSetflag = 0;

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

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

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

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

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