题目:对角线遍历
题目来源:
https://leetcode.cn/leetbook/read/array-and-string/cuxq3/
题目内容:
- 以对角线遍历的顺序,用一个数组返回m*n矩阵mat中的所有元素。
-
[[1,2,3],[4,5,6],[7,8,9]] 输出:[1,2,4,7,5,3,6,8,9]
C++代码实现
分类讨论,i % 2 == 1时,为奇数对角线,i从0开始奇数,表示对角线遍历记号。
时间复杂度:平方阶
class Solution {
public:
vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
vector<int> vec;
int m = mat.size(), n = mat[0].size();
if(mat.empty())
return vec