1. 题目
给定一个含有 M x N 个元素的矩阵(M 行,N 列),请以对角线遍历的顺序返回这个矩阵中的所有元素,对角线遍历如下图所示。
输入:
[[ 1, 2, 3 ],[ 4, 5, 6 ],[ 7, 8, 9 ]
]输出: [1,2,4,7,5,3,6,8,9]
2. 解题
- 横纵坐标的和在一条线上是一样的,每画一条线,和+1
- 向上走,i从r-1开始,否则从0开始,j = sum-i
- 只要判断位置(i,j)有效即可
class Solution {int r,c;
public:vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {if(matrix.size() == 0)return {};int i = 0, j = 0;r = matrix.size(), c = matrix[0].size();vector<int> ans;for(int k = 0; k <= r+c-1; ++k){if(k%2 == 0)//向上走{for(i = r-1; i >= 0; --i){j = k-i;if(j>=0 && j<c)ans.push_back(matrix[i][j]);}}else{for(i = 0; i < r; ++i){j = k-i;if(j>=0 && j<c)ans.push_back(matrix[i][j]);}}}return ans;}
};