LeetCode——498 对角线遍历

题目描述：

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/diagonal-traverse/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果：

代码描述：

利用每层的索引和相等

索引和为偶数，向↗遍历，

• 如果到达最后侧，则行++；
• 如果到达首行，则列++；
• 其他，行项--，列项++

索引和为奇数，向↙遍历，

• 如果到达最左侧，则行++；
• 如果到达尾行，则列++；
• 其他，行项++， 列项--

0:              (00) .................      偶数层
1:            (01)(10)   .............      奇数层
2:          (20)(11)(02)
3:        (03)(12)(21)(30)
4:      (40)(31)(22)(13)(04)
5:    (05)(14)(23)(32)(41)(50)
6:  (60)(51)................(06)

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.empty())  return res;
           
        int maxI = matrix.size();       // 行
        int maxJ = matrix[0].size();    // 列
        int j = 0;  // 列
        int i = 0;  // 行
        for(int k = 0; k < maxJ*maxI; ++k)   //  所有元素
        {
            res.push_back(matrix[i][j]);
            if((i+j)%2 == 0)        // 索引和为偶数,往↗遍历
            {
                if(j == maxJ - 1)   // 到达最右侧，下移
                    ++i;
                else if(i == 0)     // 顶部，右移
                    ++j;
                else    // 中间位置，↗移动
                {
                    --i;
                    ++j;
                }
            }
            else    // 索引和为奇数，往↙遍历
            {
                if(i == maxI - 1)   // 达到底部，右移
                    ++j;
                else if(j == 0)     // 达到最左侧，下移
                    ++i;
                else                // 往↙移动
                {
                    ++i;
                    --j;
                }
            }
        }
        return res;
    }
};