题目：对角线遍历

问题描述：

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

解决思路：

根据上面的图我们可以得到如下信息：

解决代码：

class Solution {
    public static int[] findDiagonalOrder(int[][] matrix) {
        int rowNum = matrix.length;
        int colNum = matrix[0].length;
        int[] arr = new int[rowNum * colNum];
        int[][] tran = {{-1, 1}, {1, -1}};  
        int k = 0;
        int row = 0;
        int col = 0;
        for(int i = 0; i < arr.length; i++) {
            arr[i] = matrix[row][col];
            row += tran[k][0];
            col += tran[k][1];

            // 临界情况
            if(row > rowNum - 1) {
                row = rowNum - 1;   
                col += 2;     // 在进入该代码块之前col已经减去了1，所以为了达到col+1的效果，我们需要将col加2
                k = 1 - k;    // 变化遍历方向
            }

            if(col > colNum - 1) {
                col = colNum - 1;
                row += 2;    // 在进入该代码块之前row已经减去了1，所以为了达到row+1的效果，我们需要将row加2
                k = 1 - k;
            }

            if(row < 0) {
                row = 0;
                k = 1 - k;
            }

            if(col < 0) {
                col = 0;
                k = 1 - k;
            }
        }
        return arr;
    }
}

对角线遍历

原文：https://www.cnblogs.com/syhyfh/p/12699447.html