LeetCode498. 对角线遍历

LeetCode498. 对角线遍历

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。
示例:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]

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

解释:

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

解题思路

这是看大佬写的c++代码，我转成了java代码。
（可以结合题目给的图来看）
得知：
每一趟对角线中元素的坐标（x, y）相加的和是递增的。即第一趟 x + y == 0，第二趟 x + y == 1，第三趟 x + y == 2 ……
每一趟都是 x 或 y 其中一个从大到小（每次-1），另一个从小到大（每次+1）。
例如这一趟是 x 从大到小， x 尽量取最大，当初始值超过 x 的上限时，不足的部分加到 y 上面。
例如这一趟是 x 从大到小，这一趟结束的判断是， x 减到 0 或者 y 加到上限。
这一趟是 x 从大到小，那么下一趟是 y 从大到小，循环进行。 并且方向相反时，逻辑处理是一样的，除了x，y和他们各自的上限值是相反的。

代码实现

class Solution {
    public int[] findDiagonalOrder(int[][] matrix) {
        int[] a=new int[0];
        if (matrix.length==0) return a;
        int m = matrix.length;
        int n = matrix[0].length;
        int j=0;
        int[] nums =new int[m*n];
	    boolean bXFlag = true;
	    for (int i = 0; i < m + n; i++) // 1. 坐标（x, y）相加的和是递增的
	        {
		// 2. 逻辑处理是一样的，x，y的上限值是相反的
		    int pm = bXFlag ? m : n;
		    int pn = bXFlag ? n : m;
            
		    int x = (i < pm) ? i : pm - 1; // 3. 例如这一趟是 x 从大到小， x 尽量取最大
		    int y = i - x;                 // 3. 当初始值超过 x 的上限时，不足的部分加到 y 上面

		    while (x >= 0 && y < pn) // 4. 这一趟结束的判断是， x 减到 0 或者 y 加到上限
		    {
			// 5. 方向相反时，x，y是相反的
			    nums[j]=bXFlag ? matrix[x][y] : matrix[y][x];
                j++;
			    x--; // 2. 每一趟都是 x 或 y 其中一个从大到小（每次-1）
			    y++; // 2. 另一个从小到大（每次+1）
		    }

		// 5. 循环进行
		    bXFlag = !bXFlag;
	    }
	    return nums;
    }
}

执行用时 :
3 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗 :
38.3 MB, 在所有 Java 提交中击败了98.50%的用户