矩阵学习--题目改进优化 -CSDN博客

我们来看一下针对于上一题，大佬是如何写的

大家先看这个轨迹是如何进行运行的就是一开始向右上方移动如果碰壁那就向左下角移动如果碰壁继续向右上角移动

那么设置一个result数组用来存储矩阵中的元素开始的rows和col设置为0 看有几种碰到边界的情况如果碰到的是右边界col=n-1 也就是col变成n-1了那么就调转方向而且向下移动如果碰到的是上边界 rows=0 那么就向右边移动并且调转方向不然的话就是往右上方移动就是rows-1 col+1

如果是往左下方走呢如果碰到了左方也就是col=0 那么就往下走并且调转方向如果碰到了下方 rows=2 那么就是向右走并且调转方向

向下走向右走这些使用代码表示大家应该都知道无非是列数或者行数发生变化既然这样那我们就开始尝试写代码同样的先使用正方形矩阵来写

来，我们再总结一下逻辑（控制是往右上方还是往左下方走的需要一个变量设置为flag 为true的时候往右上方走为false的时候往左下方走）

当往右上方走的时候，碰到上边界往右走方向改变碰到右边界往下走方向改变啥都没碰到就继续走

当往左下方走的时候，碰到下边界，往右走，方向改变碰到左边界往下走方向改变啥都没碰到就继续走

class Solution(object):
    def findDiagonalOrder(self, mat):
        flag=True
        n = len(mat)
        result=[]
        row=0
        col=0
        for i in range(n*n):
         result.append(mat[row][col])
         if flag:
            if row==0:
                col+=1
                flag=False
            elif col==n-1:
                row+=1
                flag=False
            else:
                row-=1
                col+=1
         else:
            if row == n-1:
                col += 1
                flag = False
            elif col == 1:
                row += 1
                flag = False
            else:
                row += 1
                col -= 1
        return result
mat=[[1,2,3],[4,5,6],[7,8,9]]
solution=Solution()
result=solution.findDiagonalOrder(mat)
print(result)

这个是我写的代码但是显示越界错误因为我把顺序弄的不好没有考虑到越界问题

if row==0:

col+=1#向右走

flag=False

elif col==n-1:

row+=1

flag=False 像这个如果row==0的时候 col也到了边界怎么办此时没办法再向右走但是对于碰到了右边界是绝对可以往下走的（这个大家最好画一下图）所以这两个的顺序要进行修改(当然我那个else部分的关于列的代码也写错了）

class Solution(object):
    def findDiagonalOrder(self, mat):
        flag=True
        n = len(mat)
        result=[]
        row=0
        col=0
        for i in range(n*n):
         result.append(mat[row][col])
         if flag:
            if col==n-1:
                row+=1
                flag=False
            elif row==0:
                col+=1
                flag=False
            else:
                row-=1
                col+=1
         else:
            if row == n-1:
                col += 1
                flag = True
            elif col == 0:
                row += 1
                flag = True
            else:
                row += 1
                col -= 1
        return result
mat=[[1,2,3],[4,5,6],[7,8,9]]
solution=Solution()
result=solution.findDiagonalOrder(mat)
print(result)

这样就对了然后再修改一下应对与m*n的情况

class Solution(object):
    def findDiagonalOrder(self, mat):
        flag=True
        m=len(mat)
        n = len(mat[0])
        result=[]
        row=0
        col=0
        for i in range(m*n):
         result.append(mat[row][col])
         if flag:
            if col==n-1:
                row+=1
                flag=False
            elif row==0:
                col+=1
                flag=False
            else:
                row-=1
                col+=1
         else:
            if row == m-1:
                col += 1
                flag = True
            elif col == 0:
                row += 1
                flag = True
            else:
                row += 1
                col -= 1
        return result
mat=[[1,2,3],[4,5,6],[7,8,9]]
solution=Solution()
result=solution.findDiagonalOrder(mat)
print(result)

同样的很好改我觉得我之前写的那个对角线变化与min（）的思路很不好这个还蛮好的而且考虑的情况也很全面比我那个好想多了

码字不易可以点赞收藏加关注吗？