力扣：对角线遍历

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

示例：

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

解析：
来自力扣题解
无论这一趟的方向如何，逻辑处理是一样的，那写两套差不多的代码就比较蛋疼，试着合并成一套。

bool bXFlag = true;
使用一个bool值标记，将用到的数据正确定位。例如：
// 5. 逻辑处理是一样的，x，y的上限值是相反的
int pm = bXFlag ? m : n;
int pn = bXFlag ? n : m;
这样就能保证逻辑计算的代码一样，数据不同。

以一趟为一个单元，那么 i 就是每一趟的和。
// 1. 坐标（x, y）相加的和是递增的
for (int i = 0; i < m + n; i++)

根据规律，就可以确定 x 和 y 的初始值。
// 3. 当初始值超过 x 的上限时，不足的部分加到 y 上面
int x = (i < pm) ? i : pm - 1;
int y = i - x;

每趟的结束条件。
// 4. 这一趟结束的判断是， x 减到 0 或者 y 加到上限
while (x >= 0 && y < pn)

一趟之中， x 和 y 变化。
// 2. 每一趟都是 x 或 y 其中一个从大到小（每次-1），另一个从小到大（每次+1）
x–; y++;

输出数字。注意根据标志位变化。
// 5. 方向相反时，x，y是相反的
nums.push_back(bXFlag ? matrix[x][y] : matrix[y][x]);

这一趟结束了，循环起来，标志位变化。
bXFlag = !bXFlag;

代码：

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) 
    {
        vector<int> nums;
        int m = matrix.size();
        if (m == 0) return nums;
        int n = matrix[0].size();
        if (n == 0) return nums;
    
        bool bXFlag = true;
        for (int i = 0; i < m + n; i++)
        {
            int pm = bXFlag ? m : n;
            int pn = bXFlag ? n : m;
    
            int x = (i < pm) ? i : pm - 1;
            int y = i - x;                

            while (x >= 0 && y < pn)
            {
                nums.push_back(bXFlag ? matrix[x][y] : matrix[y][x]);
                x--;
                y++;
            }
    
            bXFlag = !bXFlag;
        }
        return nums;
    }
};