leetcode 算法题566 (简单145) 重塑矩阵
- 题目介绍
在MATLAB中,有一个非常有用的函数 reshape,
它可以将一个矩阵重塑为另一个大小不同的新矩阵,
但保留其原始数据。
给出一个由二维数组表示的矩阵,
以及两个正整数r和c,
分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。
如果具有给定参数的reshape操作是可行且合理的,
则输出新的重塑矩阵;
否则,输出原始矩阵
- 示例
输入: nums = [[1,2], [3,4]] r = 1, c = 4
输出: [[1,2,3,4]]
解释: 行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reshape-the-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
输入: nums = [[1,2], [3,4]] r = 2, c = 4
输出: [[1,2], [3,4]]
解释: 没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。
- 说明
给定矩阵的宽和高范围在 [1, 100]。
给定的 r 和 c 都是正数。
- 解法一
/**
* @param {number[][]} nums
* @param {number} r
* @param {number} c
* @return {number[][]}
*/
var matrixReshape = function(nums, r, c) {
if(r * c !== nums[0].length * nums.length) {
return nums;
}
let i = 0, temp = [];
while(i < nums.length) {
let j = 0;
while(j < nums[i].length) {
if((i * nums[i].length + j) % c === 0) {
temp[(i * nums[i].length + j) / c] = [];
}
temp[Math.floor((i * nums[i].length + j) / c)][(i * nums[i].length + j) % c] = nums[i][j];
j++;
}
i++;
}
return temp;
};
执行用时 : 96 ms, 在所有 JavaScript 提交中击败了100.00%的用户
内存消耗 : 40.7 MB, 在所有 JavaScript 提交中击败了62.50%的用户