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; } }
}
};