LeetCode——498.对角线遍历

通过万岁！！！

• 题目：给你一个m*n的矩形，然后你按照对角线，呈s型进行遍历到一个一位数组中。
• 思路：首先我们按照层进行遍历，我们可以发现，对角线一共有n+m-1条。并且我们定义一个变量，flag，来记录对角线是右上还是左下。接下来就是确定每条对角线的开始位置，进行遍历即可。其中比较难处理的就是对角线的起始位置。
• 技巧：数组的遍历

java代码

class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int m = mat.length;
        int n = mat[0].length;
        if (m == 1) {
            return mat[0];
        }
        int[] ans = new int[n * m];
        boolean flag = true;// true表示往右上走
        // 一共有n*m-1层
        int idx = 0;
        for (int level = 0; level < n + m - 1; level++) {
            if (flag) {// 右上
                if (level < m) {// 第level行，第0列开始
                    for (int i = level, j = 0; i >= 0 && j < n; --i, ++j) {
                        ans[idx++] = mat[i][j];
                    }
                } else {// m-1行，第level-(m-1)
                    for (int i = m - 1, j = level - m + 1; i >= 0 && j < n; --i, ++j) {
                        ans[idx++] = mat[i][j];
                    }
                }
            } else {// 左下方向
                if (level < n) {// 第0行，第level列
                    for (int i = 0, j = level; i < m && j >= 0; ++i, --j) {
                        ans[idx++] = mat[i][j];
                    }
                } else {// 第level-n行 ，第n-1列
                    for (int i = level - n + 1, j = n - 1; i < m && j >= 0; ++i, --j) {
                        ans[idx++] = mat[i][j];
                    }
                }
            }
            flag = !flag;
        }
        return ans;
    }
}

• 总结：题目不是特别难，就是需要找到规律即可。