[easy]566. Reshape the Matrix

来源:互联网 发布:2017 a号黑豹数据 编辑:程序博客网 时间:2024/06/08 05:07

【原题链接】

题意:将数组转换为指定行数列数的数组。
用python刷leetcode真的是在作弊。不过人生苦短~
思路:将原数组转为一位数组,在利用分片截取list。

class Solution(object):    def matrixReshape(self, nums, r, c):        l = len(nums) * len(nums[0])        if l < r*c :           return nums        else :            rs = []            rs2 = []            for li in nums:                rs += li            for i in range(0, r) :                end = c*(i + 1)                start = c*i                rs2.append(rs[start:end:])            return rs2
原创粉丝点击