LeetCode 498. 对角线遍历

1. 题目

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

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

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

2. 解题

• 横纵坐标的和在一条线上是一样的，每画一条线，和+1
• 向上走，i从r-1开始，否则从0开始，j = sum-i
• 只要判断位置（i,j）有效即可

class Solution {
    int r,c;
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        if(matrix.size() == 0)
            return {};
        int i = 0, j = 0;
        r = matrix.size(), c = matrix[0].size();
        vector<int> ans;
        for(int k = 0; k <= r+c-1; ++k)
        {
            if(k%2 == 0)//向上走
            {
                for(i = r-1; i >= 0; --i)
                {
                    j = k-i;
                    if(j>=0 && j<c)
                        ans.push_back(matrix[i][j]);
                }
            }
            else
            {
                for(i = 0; i < r; ++i)
                {
                    j = k-i;
                    if(j>=0 && j<c)
                        ans.push_back(matrix[i][j]);
                }
            }
        }
        return ans;
    }
};