每天刷两道题——第十四天

1.1矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用原地算法。

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

原地算法（in-place algorithm）

基本上不需要额外辅助的数据结构,然而,允许少量额外的辅助变量来转换数据的算法。当算法运行时，输入的数据通常会被要输出的部分覆盖掉。不是原地算法有时候称为非原地（not-in-place）或不得其所（out-of-place）。

代码

    def setZeroes(self,matrix):
        m,n=len(matrix),len(matrix[0])
        row,col=[False]*m,[False]*n
        for i in range(m):
            for j in range(n):
                if matrix[i][j]==0:
                    row[i]=col[j]=True  #把0所在行和列标记为True

        for i in range(m):
            for j in range(n):
                if row[i] or col[j]:
                    matrix[i][j]=0
        return matrix

1.2螺旋矩阵

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 返回矩阵中的所有元素。

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

代码

    def spiralOrder(self,matrix):
        m,n=len(matrix),len(matrix[0])
        upper,left,right,down=0,0,n-1,m-1 #四个位置边界的下标
        res=[]
        while True:
            for i in range(left,right+1): #向右移动到最右
                res.append(matrix[upper][i])
            upper+=1 #缩小上边界，去掉已经遍历过的行
            if upper>down:
                break
            for i in range(upper,down+1): #向下移动到最底部
                res.append(matrix[i][right])
            right-=1 #缩小右边界，去掉已经遍历过的列
            if right<left:
                break
            for i in range(right,left-1,-1):  #往左
                res.append(matrix[down][i])
            down-=1 #缩小下边界，去掉遍历过的行
            if down<upper:
                break
            for i in range(down,upper-1,-1):  #向上
                res.append(matrix[i][left])
            left+=1 #缩小左边界，去掉已经遍历过的列
            if left>right:
                break
        return res

python的range函数基础用法

#range(start, stop, step)=range(初值, 终值, 步长)
#取值范围是[start,stop）
#range()函数只适用于整数，步长可以为正值也可以为负值

for i in range(4):  
    print(i)#0 1 2 3

for i in range(1,4):
    print(i) #1,2,3

for i in range(1,15,4):
    print(i) #1,5,9,13

for i in range(4,1,-1):
	print(i) #4,3,2

n=list(range(4))
print(n) #[0,1,2,3]

参考代码