In MATLAB, there is a very useful function called ‘reshape’, which can reshape a matrix into a new one with different size but keep its original data.<\/p>\n
You’re given a matrix represented by a two-dimensional array, and two\u00a0positive<\/b>\u00a0integers\u00a0r<\/b>\u00a0and\u00a0c<\/b>\u00a0representing the\u00a0row<\/b>\u00a0number and\u00a0column<\/b>\u00a0number of the wanted reshaped matrix, respectively.<\/p>\n
The reshaped matrix need to be filled with all the elements of the original matrix in the same\u00a0row-traversing<\/b>\u00a0order as they were.<\/p>\n
If the ‘reshape’ operation with given parameters is possible and legal, output the new reshaped matrix; Otherwise, output the original matrix.<\/p>\n
Example 1:<\/b><\/p>\n Example 2:<\/b><\/p>\n Note:<\/b><\/p>\n Python<\/strong><\/p>\nInput:<\/b> \r\nnums = \r\n[[1,2],\r\n [3,4]]\r\nr = 1, c = 4\r\nOutput:<\/b> \r\n[[1,2,3,4]]\r\nExplanation:<\/b>\r\nThe row-traversing<\/b> of nums is [1,2,3,4]. The new reshaped matrix is a 1 * 4 matrix, fill it row by row by using the previous list.\r\n<\/pre>\n
Input:<\/b> \r\nnums = \r\n[[1,2],\r\n [3,4]]\r\nr = 2, c = 4\r\nOutput:<\/b> \r\n[[1,2],\r\n [3,4]]\r\nExplanation:<\/b>\r\nThere is no way to reshape a 2 * 2 matrix to a 2 * 4 matrix. So output the original matrix.\r\n<\/pre>\n
\n
\r\nclass Solution(object):\r\n def matrixReshape(self, nums, r, c):\r\n """\r\n :type nums: List[List[int]]\r\n :type r: int\r\n :type c: int\r\n :rtype: List[List[int]]\r\n """\r\n l = [[]]\r\n if not nums:\r\n return nums\r\n if len(nums) * len(nums[0]) != r * c:\r\n return nums\r\n else:\r\n l = [[] for i in range(r)]\r\n count = 0\r\n for nums_r in nums:\r\n for nums_c in nums_r:\r\n a_r = int(count \/ c)\r\n l[a_r].append(nums_c)\r\n count += 1\r\n return l\r\n \r\n<\/pre>\n","protected":false},"excerpt":{"rendered":"