码迷,mamicode.com
首页 > 其他好文 > 详细

Reshape the Matrix

时间:2017-09-21 20:54:05      阅读:125      评论:0      收藏:0      [点我收藏+]

标签:img   元素   ima   matrix   ==   obj   ret   src   col   

    这道题为简单题

  题目:

    技术分享

 

  思路:

    首先判断该列表是否满足‘reshape’,计算二维列表的长度和宽度并且判断是否长度宽度相乘等于r*c,如果相等就遍历列表每个元素将其加入新列表中,否则就返回原二维列表

  代码:

 1 class Solution(object):
 2     def matrixReshape(self, nums, r, c):
 3         """
 4         :type nums: List[List[int]]
 5         :type r: int
 6         :type c: int
 7         :rtype: List[List[int]]
 8         """
 9         a = len(nums)
10         b = len(nums[0])
11         d = [[None] * c for _ in xrange(r)]
12         if a * b == r * c:
13             for i in range(r*c):
14                 d[i//c][i%c]=nums[i//b][i%b]
15             return d
16         else: return nums

 

Reshape the Matrix

标签:img   元素   ima   matrix   ==   obj   ret   src   col   

原文地址:http://www.cnblogs.com/liuxinzhi/p/7570694.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!