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的插入元素,使用push_back();
  • 二维数组有可能行长度为0,列长度为0,要分开讨论,先讨论行不为0的情况,例如测试点[],martrix为NULL,那么martrix[0]的定义就会有错误。
class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        vector<int> result;
        int m=matrix.size();
        if(!m)return result;
        int n=matrix[0].size();
        if(!n)return result;
        int r,c,k;
        r=c=k=0;
        int d[2][2]={{-1,1},{1,-1}};
        for(int i=0;i<m*n;i++){
            result.push_back(matrix[r][c]);
            r+=d[k][0];
            c+=d[k][1];
            if(c>n-1){
                c=n-1;
                r+=2;
                k=1-k;
            }
            if(r<0){
                r=0;
                k=1-k;
            }
            if(r>m-1){
                r=m-1;
                c+=2;
                k=1-k;
            }
            if(c<0){
                c=0;
                k=1-k;
            }
        }
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值