leetcode 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 。

这道题的思路就是判断一下边界，用一个up变量记录方向，碰到边界改变方向，并且改变相应的坐标值i，j
AC代码如下：

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {

        vector<int> vec;
        if(matrix.empty()) return vec;
        int i=0,j=0,m=matrix.size()-1,n=matrix[0].size()-1,up=1;

        while(i!=m||j!=n){
            //printf("%d+%d ",i,j);
            vec.push_back(matrix[i][j]);
            if(i==0&&up==1){
                up=0;
                if(j==n)  i++;
                else j++;   
                continue;
            }
            if(j==0&&up==0){
                up=1;
                if(i==m) j++;
                else i++;
                continue;
            }
            if(i==m&&up==0){
                up=1;
                j++;
                continue;
            }
            if(j==n&&up==1){
                up=0;
                i++;
                continue;
            }
            if(up==0){
                i++;
                j--;
            }
            else{
                i--;
                j++;
            }
        }
        vec.push_back(matrix[i][j]);
        return vec;
    }
};