python(leetcode)498. 对角线遍历

这题难度中等，记录下思路

第一个会超时，

第二个：思想是按斜对角线行进行右下左上交替遍历，

def traverse(matrix):
    n=len(matrix)-1
    m=len(matrix[0])-1
    result=[]
    for i in range(m+n+1):
            if(i % 2 == 0):
                for j in range(i, -1, -1):
                    x=j
                    y=i-x
                    if x <= n and y <= m:
                        result.append(matrix[x][y])
                    # elif y > m:
                    #     break
                    else:
                        continue
            else:
                for j in range(i, -1, -1):
                    y=j
                    x=i-y
                    if x <= n and y <= m:
                        result.append(matrix[x][y])
                    # elif x > n:
                    #     break
                    else:
                        continue
    return result

if __name__ == '__main__':
    ma=[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
print(traverse(ma))

11

def diagonial(matrix):
    m=len(matrix)
    n=len(matrix[0])
    #思想是按斜对角线行进行遍历
    #遍历的方向右下，左上交替进行
    line=0
    res=[]
    tem=[]
    is_sure=True
    #对每一行进行循环 斜对角线有m+n-1条
    while line<(m+n-1):
    #设置起点
        if(line<m):
            x=line
            y=0
        else:
            x=m-1
            y=line-m-1
        #每行循环几次
        #因为都是从右上开始，所以设置条件x>0,y<n
        while x>=0 and y <n:
            # 存储斜行的元素
            tem.append(matrix[x][y])
            x-=1
            y+=1
        #如果是左下开始，则反向添加
        if(is_sure):
            tem.reverse()
        for num in tem:
            res.append(num)
        line+=1
    return res


if __name__ == '__main__':
    matrix=[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
    print(diagonial(matrix))