leetcode 498. 对角线遍历（矩形输出对角线遍历）

题目

1. 想往上去越界了：看是否可以往右去(y<m-1)，可以的话往右，否则往下
2. 想往下去越界了：看是否可以往下去(x<n-1)，可以的话往下，否则往右

记住要每次越界之后，要变方向，flag^=1;

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& ma) {
        vector<int>ans;
        int n=ma.size();
        if(!n) return ans;
        int m=ma[0].size();
        int x=0,y=0;ans.push_back(ma[x][y]);
        int flag=0;
        while(!(x==n-1&&y==m-1)){
            int nx=flag?x+1:x-1,ny=flag?y-1:y+1;
            if(!(nx>=0&&nx<n&&ny>=0&&ny<m)){//越界了 就该变方向了
                if(!flag){
                    if(y<m-1) nx=x,ny=y+1;
                    if(y==m-1) nx=x+1,ny=y;
                }else{
                    if(x<n-1) nx=x+1,ny=y;
                    if(x==n-1) nx=x,ny=y+1;
                }
                flag^=1;
            }
            ans.push_back(ma[x=nx][y=ny]);
        }
        return ans;
    }
};