Java实现 LeetCode 498 对角线遍历

迷南。 2023-07-17 14:05 7阅读 0赞

## 498. 对角线遍历 ##

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

示例:

输入:  
\[  
\[ 1, 2, 3 \],  
\[ 4, 5, 6 \],  
\[ 7, 8, 9 \]  
\]

输出: \[1,2,4,7,5,3,6,8,9\]

解释:

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70]

说明:

给定矩阵中的元素总数不会超过 100000 。

PS：  
对角线的特点就是x+y相等

class Solution { 
       public int[] findDiagonalOrder(int[][] matrix) { 
        if (matrix == null || matrix.length == 0) { 
            return new int[]{ };
        }
        int r = 0, c = 0;
        int row = matrix.length, col = matrix[0].length;
        int[] res = new int[row * col];
        for (int i = 0; i < res.length; i++) { 
            res[i] = matrix[r][c];
            // r + c 即为遍历的层数，偶数向上遍历，奇数向下遍历
            if ((r + c) % 2 == 0) { 
                if (c == col - 1) { 
                    // 往右移动一格准备向下遍历
                    r++;
                } else if (r == 0) { 
                    // 往下移动一格准备向下遍历
                    c++;
                } else { 
                    // 往上移动
                    r--;
                    c++;
                }
            } else { 
                if (r == row - 1) { 
                    // 往右移动一格准备向上遍历
                    c++;
                } else if (c == 0) { 
                    // 往上移动一格准备向上遍历
                    r++;
                } else { 
                    // 往下移动
                    r++;
                    c--;
                }
            }
        }
        return res;
    }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70]: /images/20230528/18c0ef849b73454dbd22669e5c8e6d74.png