56、重塑矩阵

在 MATLAB 中,有一个非常有用的函数 reshape ,它可以将一个 m x n 矩阵重塑为另一个大小不同(r x c)的新矩阵,但保留其原始数据。
给你一个由二维数组 mat 表示的 m x n 矩阵,以及两个正整数 r 和 c ,分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。
如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。
示例 1:
输入:mat = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]
示例 2:
输入:mat = [[1,2],[3,4]], r = 2, c = 4
输出:[[1,2],[3,4]]
思路:那么题目需要我们做的事情相当于:将二维数组nums 映射成一个一维数组;将这个一维数组映射回 r 行 cc列的二维数组。我们当然可以直接使用一个一维数组进行过渡,但我们也可以直接从二维数组nums 得到 r 行 cc列的重塑矩阵:
设 nums 本身为 m 行 n 列,如果 mn = rc那么二者包含的元素个数不相同,因此无法进行重塑;否则,对于 x∈[0,mn),第 x 个元素在nums 中对应的下标为 (x / n,x % n),而在新的重塑矩阵中对应的下标为(x / c,x % c)。我们直接进行赋值即可。

int R=mat.length;
        int C = mat[0].length;
        if(R*C!=r*c){
            return mat;
        }
        int[][] mat2 = new int[r][c];
        for (int i = 0; i < R * C; i++) {
            mat2[i / c][i % c] = mat[i / C][i % C];
        }
        System.out.println(mat2.toString());
        return mat2;
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值