对角线遍历
题目描述:给定一个含有 M x N 个元素的矩阵(M 行,N 列),请以对角线遍历的顺序返回这个矩阵中的所有元素,对角线遍历如下图所示。
示例:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,4,7,5,3,6,8,9]
解释:
说明:
给定矩阵中的元素总数不会超过 100000 。
解题思路:这一题刚开始的思路,就比较容易想到螺旋矩阵那一题,借鉴那一题的思路,可以像那一题一样,模拟走的方向,确定下一个点的位置,这一题不同于螺旋矩阵的是,此题的方向只有两个,右上和左下,定义一个二维数组即可,然后碰到边界,就要转向,转向的处理,一定要手动模拟一下,才不会出错。
- vector的插入元素,使用push_back();
- 二维数组有可能行长度为0,列长度为0,要分开讨论,先讨论行不为0的情况,例如测试点[],martrix为NULL,那么martrix[0]的定义就会有错误。
class Solution {
public:
vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
vector<int> result;
int m=matrix.size();
if(!m)return result;
int n=matrix[0].size();
if(!n)return result;
int r,c,k;
r=c=k=0;
int d[2][2]={{-1,1},{1,-1}};
for(int i=0;i<m*n;i++){
result.push_back(matrix[r][c]);
r+=d[k][0];
c+=d[k][1];
if(c>n-1){
c=n-1;
r+=2;
k=1-k;
}
if(r<0){
r=0;
k=1-k;
}
if(r>m-1){
r=m-1;
c+=2;
k=1-k;
}
if(c<0){
c=0;
k=1-k;
}
}
return result;
}
};