5.LC 零矩阵(中等)

面试题 01.08. 零矩阵 - 力扣（LeetCode）

思想:

法一:
利用两个集合分别储存要清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] = 0for x in lie:for i in range(n):matrix[i][x] = 0

相似题

73. 矩阵置零 - 力扣（LeetCode）

6.LC 对角线遍历(中等，学习)

498. 对角线遍历 - 力扣（LeetCode）

思想:

1.n行m列矩阵,因为是按照对角线遍历，但是分析可得有两种对角线遍历方式，从左下到右上和从右上到左下，是由第i条对角线的奇偶决定的，所以按照对角线遍历，$i\in [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 + 1y = i if i < m else m - 1while x < n and y >= 0:res.append(mat[x][y])x += 1y -= 1else:y = 0 if i < n else i - n + 1x = i if i < n else n - 1while y < m and x >= 0:res.append(mat[x][y])y += 1x -= 1return resclass 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+1y=i if i<m else m-1while x<n and y>=0:res.append(mat[x][y])x+=1y-=1else:y=0 if i<n else i-n+1x=i if i<n else n-1while y<m and x>=0:res.append(mat[x][y])y+=1x-=1return res

1.没有三目运算符，要写成x=0 if i<m else i-m+1
2.没有++运算符，要写成x+=1