498.Diagonal Traverse-M

题目内容

给定一个M*N大小的矩阵（M行，N列），要求返还一个以下图所示方法得到的一个一维数组

这道题从题意来看，很简洁，要求按照对角线以S型遍历整个数组，从而得到一个特定顺序的一维数组，如果是人来完成这个任务，恐怕五六岁的小朋友都能轻易做到。但要怎么让电脑完成这件事呢？你需要注意到这几个点：

• 上图中任一对角线上的所有点， 横纵坐标之和是相同的。
• 横纵坐标之和的奇偶性决定了对角线应该向右上方遍历，还是应该向左下方遍历，即横纵坐标的变化方法；当和为奇数时，向左下方遍历；和为偶数时，向右上方遍历。
• 最重要的一点，每次对角线遍历到达上下左右四个边界后，应该怎么选择起点以及更新横纵坐标变换的方式，下面我将结合代码对这一点加以阐述。

class Solution {public:    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {        vector<int> result;        if (matrix.size() == 0) return result;        // 初始遍历方向为右上方        int row = matrix.size(), col = matrix[0].size();        int sum = 0, x = 0, y = 0, dx = -1, dy = 1;        while (x < row && y < col) {            result.push_back(matrix[x][y]);            x += dx;            y += dy;            if (x < 0 || y < 0 || x >= row || y >= col) {                ++sum;                if (x < 0) {                    if (y < 0) {                        // 这种情况不会发生                    } else if (y >= col) {                        // 遍历接触到右侧边界线                        x += 2;                        y = col - 1;                    } else {                        // 遍历接触到上侧边界线                        x = 0;                    }                } else if (x >= row) {                    x = r - 1;                    if (y < 0) {                        // 遍历接触到下侧边界线                        y += 2;                    } else if (y >= col) {                        // this condition won't happen                    } else {                        // 遍历接触到下侧边界线                        y += 2;                    }                } else {                    if (y < 0) {                        // 遍历接触到左侧边界线                        y = 0;                    } else if (y >= col) {                        // 遍历接触到右侧边界线                        y = col - 1;                        x += 2;                    }                }                // 根据奇偶性确定下一次的遍历方向                if (sum % 2 == 1) {                    dx = 1;                    dy = -1;                } else {                    dx = -1;                    dy = 1;                }            }        }        return result;    }};