566. Reshape the Matrix

https://leetcode.com/problems/reshape-the-matrix/

Reshape the Matrix - LeetCode

 

from typing import List


class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        flat = tuple(each for row in mat for each in row)
        if len(flat) != r * c:
            return mat
        return [flat[i: i + c] for i in range(0, len(flat), c)]

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        flat = tuple(each for row in mat for each in row)
        return mat if len(flat) != r * c else [flat[i: i + c] for i in range(0, len(flat), c)]

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        return mat if len(flat := tuple(each for row in mat for each in row)) != r * c else [flat[i: i + c] for i in range(0, len(flat), c)]