​LeetCode刷题实战498：对角线遍历

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
      //处理特殊的情形
        if(matrix.empty()||matrix[0].empty()){
            return {};
        }
        //初始化相关变量
        vector<int> res;
        int m=matrix.size();
        int n=matrix[0].size();
    //遍历数组
        int i=0;
        while(i<m+n){
          //从左下到右上的遍历
            int x=i<m?i:m-1;
            int y=i-x;
            while(x>=0&&y<n){//该趟遍历的终止位置
                res.push_back(matrix[x][y]);
                --x;
                ++y;
            }
            ++i;//跳转下一趟遍历

            if(i==m+n){//处理可能的越界
                break;
            }
            
      //从右上到左下
            y=i<n?i:n-1;
            x=i-y;
            while(y>=0&&x<m){//该趟遍历的终止位置
                res.push_back(matrix[x][y]);
                ++x;
                --y;
            }
            ++i;
        }
        return res;
    }
};