RE:0 从零开始的flag生活(力扣,数组篇,498

基本思路:由于对角线性质,对角线划出来最多只有长加宽减1条(0到n+m-1),而且偶数对角线是朝上的,奇数对角线是朝下的,朝上时维护行不过顶且列不过右,朝下时维护列不过左,行不过底即可。代码如下:

class Solution:
    def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]: 
        k=[]
        t,j=0,0
        for i in range(len(mat)+len(mat[0])-1):
            if i%2==0:
                if t==len(mat):
                    t-=1
                    j+=1
                while t>=0 and len(mat[0])-1>=j:
                    k.append(mat[t][j])
                    t-=1
                    j+=1
                t+=1
            if i%2==1:
                if j==len(mat[0]):
                    j-=1
                    t+=1
                while len(mat)-1>=t and j>=0:
                    k.append(mat[t][j])
                    t+=1
                    j-=1
                j+=1
        return k

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值