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 。

这道题的思路就是判断一下边界,用一个up变量记录方向,碰到边界改变方向,并且改变相应的坐标值i,j
AC代码如下:

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {

        vector<int> vec;
        if(matrix.empty()) return vec;
        int i=0,j=0,m=matrix.size()-1,n=matrix[0].size()-1,up=1;

        while(i!=m||j!=n){
            //printf("%d+%d ",i,j);
            vec.push_back(matrix[i][j]);
            if(i==0&&up==1){
                up=0;
                if(j==n)  i++;
                else j++;   
                continue;
            }
            if(j==0&&up==0){
                up=1;
                if(i==m) j++;
                else i++;
                continue;
            }
            if(i==m&&up==0){
                up=1;
                j++;
                continue;
            }
            if(j==n&&up==1){
                up=0;
                i++;
                continue;
            }
            if(up==0){
                i++;
                j--;
            }
            else{
                i--;
                j++;
            }
        }
        vec.push_back(matrix[i][j]);
        return vec;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值