【python】LeetCode刷题 566. Reshape the Matrix

最新推荐文章于 2022-08-28 14:48:44 发布

原创最新推荐文章于 2022-08-28 14:48:44 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种将二维列表重塑为指定维度的一维列表的算法。该算法首先将输入的二维列表展平为一维列表，然后根据指定的行数和列数重新组织这个一维列表，形成新的二维列表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution(object):
    def matrixReshape(self, nums, r, c):
        nums_row=len(nums)
        nums_col=len(nums[0])
        if nums_row*nums_col==r*c:
            tmp=[num for row in nums for num in row]
            col_tmp=[]
            for i in range(0,r):
                row_tmp = []
                for j in range(0,c):
                    row_tmp.append(tmp[i*c+j])
                col_tmp.append(row_tmp)
            return col_tmp
        else:
            return nums
nums =[[1, 2],[3, 4]]
so=Solution()
print so.matrixReshape(nums,1,4)

采用的方法很笨，基本思路是先把列表展开成一个列表，
然后再根据题目要求的维数，把一维列表转换成要求的列表数目。