給出一個(gè)由二維數(shù)組表示的矩陣，以及兩個(gè)正整數(shù)r和c，分別表示想要的重構(gòu)的矩陣的行數(shù)和列數(shù)。重構(gòu)后的矩陣需要將原始矩陣的所有元素以相同的行遍歷順序填充。如果具有給定參數(shù)的reshape操作是可行且合理的，則輸出新的重塑矩陣；否則，輸出原始矩陣。

例子:

輸入
nums =
[[1,2],
[3,4]]
r = 1, c = 4

輸出
[[1,2,3,4]]

解釋
行遍歷nums的結(jié)果是 [1,2,3,4]。新的矩陣是 1 * 4 矩陣, 用之前的元素值一行一行填充新矩陣。

輸入
nums =
[[1,2],
[3,4]]
r = 2, c = 4
輸出
[[1,2],
[3,4]]

解釋
沒有辦法將 2 * 2 矩陣轉(zhuǎn)化為 2 * 4 矩陣。 所以輸出原矩陣。

思路:

暴力法

題意, 不難理解, 我們可以按照題意機(jī)械翻譯得到

1.判斷nums元素個(gè)數(shù)與r,c元素個(gè)數(shù)是否相等
① 不等, 直接排除
② 相等, 循環(huán)插入新數(shù)組

未翻譯版

class Solution {
    func matrixReshape(_ nums: [[Int]], _ r: Int, _ c: Int) -> [[Int]] {
        
        let sum = nums[0].count * nums.count, target = r * c
        if sum != target {
            return nums
        }
        
        var result:[[Int]] = [], temp:[Int] = []
        
        for arr in nums {
            for j in arr {
                temp.append(j)
                if temp.count == c {
                    result.append(temp)
                    temp.removeAll()
                }
            }
        }
        return result 
    }
}

翻譯版

class Solution {
    func matrixReshape(_ nums: [[Int]], _ r: Int, _ c: Int) -> [[Int]] {
        
        // 得到nums元素個(gè)數(shù), r,c 新矩陣元素個(gè)數(shù)
        let sum = nums[0].count * nums.count, target = r * c
        // 不相等直接返回
        if sum != target {
            return nums
        }
        
        // 相等, 遍歷數(shù)組
        // temp 為新數(shù)組每一個(gè)行容器
        var result:[[Int]] = [], temp:[Int] = []
        for arr in nums {
            for j in arr {
                // j為nums中從左到右依次循環(huán)的元素
                // temp添加 j
                temp.append(j)
                // 如果temp"滿了", 則換行繼續(xù)添加
                if temp.count == c {
                    result.append(temp)
                    temp.removeAll()
                }
            }
        }
        // 返回
        return result 
    }
}

題目來源：力扣（LeetCode） 感謝力扣爸爸 :)
IOS 算法合集地址