力扣数据结构入门第4天打卡

566重塑矩阵

 def matrixReshape(self, mat, r, c):
        """
        :type mat: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        temp=[]
        for i in range(len(mat)):
                for j in range(len(mat[0])):
                    temp.append(mat[i][j])
        n=0
        if r*c == len(mat)*len(mat[0]):
            res=[[0] * c for _ in range(r)]
            for i in range(r):
                for j in range(c):
                    res[i][j]=temp[n]
                    n+=1
            return res
        return mat

118杨辉三角

def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        row=[]
        yhsj=[]
        for i in range(1,numRows+1):
            yhsj.append([1] * i)

        
        if numRows>=3:
          for i in range(2,numRows):  
              for j in range(1,i):
                   yhsj[i][j]=yhsj[i-1][j-1]+yhsj[i-1][j]
        return yhsj