498对角线遍历

对角线遍历

题目描述：给定一个含有 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;
    }
};