5.LC 零矩阵(中等)
思想:
法一 :
利用两个集合分别储存要清0的行和列索引
另外两种原地优化空间的做法暂时不是目前刷题目标，故不考虑
代码
c++:
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
set<int> hang;
set<int> lie;
int n = matrix.size(), m = matrix[0].size();
for (int i = 0; i < n; ++i) {
for (int j = 0; j < m; ++j) {
if (matrix[i][j] == 0) {
hang.insert(i);
lie.insert(j);
}
}
}
for (const auto& x : hang) {
for (int j = 0; j < m; ++j) {
matrix[x][j] = 0;
}
}
for (const auto& x : lie) {
for (int i = 0; i < n; ++i) {
matrix[i][x] = 0;
}
}
}
};python:
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
hang = set()
lie = set()
n, m = len(matrix), len(matrix[0])
for i in range(n):
for j in range(m):
if matrix[i][j] == 0:
hang.add(i)
lie.add(j)
for x in hang:
for j in range(m):
matrix[x][j] = 0
for x in lie:
for i in range(n):
matrix[i][x] = 0相似题
6.LC 对角线遍历(中等，学习)
思想:
1.n行m列矩阵,因为是按照对角线遍历，但是分析可得有两种对角线遍历方式，从左下到右上和从右上到左下，是由第i条对角线的奇偶决定的，所以按照对角线遍历， i ∈ [ 0 , n + m − 1 ) i \in [0,n+m-1) i∈[0,n+m−1)
2.对于偶数对角线，是从左下到右上的遍历顺序，但又会出现遍历初始位置不同的情况，故再分类
- i<n,说明在左边界，初始位置(i,0)
- i>=n,说明在下边界，初始位置(n-1,i-(n-1))即(n-1,i-n+1)
3.奇数对角线，是从右上到左下的遍历顺序,对偶式即可 - i<m,说明在上边界，初始位置(0,i)
- i>=m,说明在右边界，初始位置(i-m+1,m-1)
代码
c++:
class Solution {
public:
vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
vector<int> res;
int n = mat.size(), m = mat[0].size();
for (int i = 0; i < n + m - 1; ++i) {
// 右上到左下
if (i % 2) {
int x = i < m ? 0 : i - m + 1;
int y = i < m ? i : m - 1;
while (x < n && y >= 0) {
res.emplace_back(mat[x][y]);
x++;
y--;
}
}
// 左下到右上
else {
int x = i < n ? i : n - 1;
int y = i < n ? 0 : i - n + 1;
while (y < m && x >= 0) {
res.emplace_back(mat[x][y]);
x--;
y++;
}
}
}
return res;
}
};python:
class Solution:
def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:
n, m = len(mat), len(mat[0])
res = []
for i in range(n + m - 1):
if i % 2 == 1:
x = 0 if i < m else i - m + 1
y = i if i < m else m - 1
while x < n and y >= 0:
res.append(mat[x][y])
x += 1
y -= 1
else:
y = 0 if i < n else i - n + 1
x = i if i < n else n - 1
while y < m and x >= 0:
res.append(mat[x][y])
y += 1
x -= 1
return res
class Solution:
def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:
n,m=len(mat),len(mat[0])
res=[]
for i in range(n+m-1):
if i%2==1:
x=0 if i<m else i-m+1
y=i if i<m else m-1
while x<n and y>=0:
res.append(mat[x][y])
x+=1
y-=1
else:
y=0 if i<n else i-n+1
x=i if i<n else n-1
while y<m and x>=0:
res.append(mat[x][y])
y+=1
x-=1
return res1.没有三目运算符，要写成x=0 if i<m else i-m+1
2.没有++运算符，要写成x+=1