单向遍历

参考答案：https://leetcode.cn/problems/diagonal-traverse/solution/tong-su-yi-dong-by-lfool-eozv/ 题目不难理解，就是边界问题很难弄。这个方法应该是最好理解的了，就是可能性能不好…

  1. int m, n;
  2. ArrayList<Integer> arrayList = new ArrayList();
  4. public int[] findDiagonalOrder(int[][] mat) {
  5.     m = mat.length;
  6.     n = mat[0].length;
  7.     boolean dir = true;//true表示要反转
  9.     //遍历第一行
  10.     for (int j = 0; j < n; j++) {
  11.         helper(mat, 0, j, dir);
  12.         dir = !dir;
  13.     }
  15.     //遍历最后一列
  16.     for (int i = 1; i < m; i++) {
  17.         helper(mat, i, n - 1, dir);
  18.         dir = !dir;
  19.     }
  21.     int[] ans = new int[m * n];
  22.     for (int i = 0; i < m * n; i++) {
  23.         ans[i] = arrayList.get(i);
  24.     }
  25.     return ans;
  26. }
  28. //沿着左下角的方向遍历
  29. public void helper(int[][] mat, int x, int y, boolean dir) {
  30.     ArrayList<Integer> list = new ArrayList<>();
  31.     for (int i = x, j = y; i < m && j >= 0; i++, j--) {
  32.         list.add(mat[i][j]);
  33.     }
  35.     if (dir) {
  36.         Collections.reverse(list);
  37.     }
  38.     arrayList.addAll(list);
  39. }