给定一个含有 M x N 个元素的矩阵(M 行,N 列),请以对角线遍历的顺序返回这个矩阵中的所有元素,对角线遍历如下图所示。
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,4,7,5,3,6,8,9]
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;
}
};