LeetCode4.Array and String — Diagonal Traverse 对角线遍历

Posted 2021-12-08 hu-19941213

Given a matrix of M x N elements (M rows, N columns), return all elements of the matrix in diagonal order as shown in the below image.

Example:

Input:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
Output:  [1,2,4,7,5,3,6,8,9]
Explanation:

          

Note:

The total number of elements of the given matrix will not exceed 10,000.

这一题我写了一下午（看在自己刚入门和智商的份上，原谅自己）

本来打算放弃了，后来强行在纸上画图，找规律，写逻辑，最终把代码编了出来，努力和坚持还是有回报的 O(∩_∩)O哈哈~

总体思路：把二维数组看成一个矩形，对应的四条边列出相应的四条判断规则，当遍历的元素“碰”（想象一下为一个会动的点在一个矩形中移动）到矩形的边时，根据规则改变方向。

从数组第一个元素开始，最后一个元素为结束。具体解释放在代码的注释里，就不多说了，脑子快炸了。。。

class Solution {
public:
vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
    if (matrix.size()==0) return{}; //判断数组是否为空
    vector<int> result; //存放结果的数组
    int M = matrix.size();//获取数组的行数 
    int N = matrix[0].size();//获取数组的列数
    int  m=0, n = 0;        //两个计数器，分别对应M、N
    result.push_back(matrix[m][n]);  //将matrix[0][0]加入到数组中.
    while (!(m==M-1 && n==N-1)) { //当遍历到最后一个元素matrix[M][N]时,循环结束.
            if (m == 0 && n < N-1) {  //请把二维数组脑补成一个矩形，这里判断的是矩形的上边规则，即当元素遍历到矩形的上边.
                result.push_back(matrix[m][++n]);
                while (n >0&&m<M-1)/右移一位，然后向左下方遍历.
                {
                    m++; n--;
                    result.push_back(matrix[m][n]);
                }
            }
            else if (n == 0 && m < M-1) //矩形的左边.即当元素遍历到矩形的最左边.
            {
                result.push_back(matrix[++m][n]);
                while (m >0&&n<N-1) //下移一位，然后向右上方遍历.
                {
                    m--; n++;
                    result.push_back(matrix[m][n]);
                }
            }
            else if ( n == N - 1 && m != M - 1) //矩形的右边.即当元素即将遍历到最右边.
            {
                result.push_back(matrix[++m][n]);
                while (m < M - 1&&n>0) //下移一位,然后向左下方遍历.
                {
                    m++; n--;
                    result.push_back(matrix[m][n]);
                }
            }
            else if ( m == M - 1 && n != N-1)//矩形的下边.即当元素遍历到最下边.
            {
                result.push_back(matrix[m][++n]);
                while (n <N-1&&m>0) //右移一位，然后向右上方遍历.
                {
                    m--; n++;
                    result.push_back(matrix[m][n]);
                }
            }
    }
    return result;
 }
};