重塑矩阵

在 MATLAB 中,有一个非常有用的函数 reshape ,它可以将一个 m x n 矩阵重塑为另一个大小不同(r x c)的新矩阵,但保留其原始数据。

给你一个由二维数组 mat 表示的 m x n 矩阵,以及两个正整数 r 和 c ,分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。

如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。

/**
 * @param {number[][]} mat
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */
const matrixReshape = (mat, r, c) => {
    const arr = mat.join().split(',')
    if(arr.length !== r * c ) return mat
    const res = []
    for(let i = 0; i < r; i++){
        res.push(arr.slice(i * c,(i + 1) * c))
    }
    return res
};

  

posted @ 2023-02-05 00:17  671_MrSix  阅读(8)  评论(0编辑  收藏  举报