Leetcode 498. 对角线遍历 C++

Leetcode 498. 对角线遍历

题目

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

示例:

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

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

解释:

说明:

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

题解

按照上对角、下对角进行遍历，利用好行数+列数=第几个对角。详细过程见代码

代码

vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        if(matrix.empty() || matrix[0].empty()) return {};
        int m = matrix.size(), n = matrix[0].size();
        vector<int> ans;
        int change=1;
        for(int i=0; i<m+n; i++){	
            if(change > 0){		//上对角
                int x = i>=m? m-1:i;		//不能超过最后一行
                int y = i-x;
                while(x>=0 && y<n){
                    ans.push_back(matrix[x][y]);
                    x--;
                    y++;
                }
            }else{			//下对角
                int y = i>=n? n-1:i;		//不能超过最后一列
                int x = i-y;
                while(y>=0 && x<m){
                    ans.push_back(matrix[x][y]);
                    x++;
                    y--;
                }
            }
            change = -change;
        }
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/diagonal-traverse
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。