听风是风

学或不学,知识都在那里,只增不减。

导航

JS LeetCode 566. 重塑矩阵题解分析,数组降维的几种方式

壹 ❀ 引

今天是过完年到公司的第二天,年前因为封版,到今天我们小组是第一个发版的组,然后就各种踩坑,到现在还在公司等运维解决jenkins问题,闲着也是闲着,做一道算法题,简单记录下解题思路,本题来自LeetCode566. 重塑矩阵,描述如下:

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

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

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

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

示例 1:

输入: 
nums = 
[[1,2],
[3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

示例 2:

输入: 
nums = 
[[1,2],
[3,4]]
r = 2, c = 4
输出: 
[[1,2],
[3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

注意:

  1. 给定矩阵的宽和高范围在 [1, 100]。
  2. 给定的 r 和 c 都是正数。

贰 ❀ 简单的题解

题目不难理解,给定一个二维数组,将其转换成一个包含r个数组,且每个数组包含c个元素的新数组。比如例子1中,就是将原数组转为包含1个数组且数组元素为4个的新数组。

需要注意的是,也可能存在转换失败的情况,比如例子2,原数组一共就只有4个元素,而期望是转成包含2个数组,且每个数组包含4个元素的新数组,数量超过了原数组元素数量,转换失败时期望返回原数组。

因此我们可以得知,不管怎么转换,转换之后的数组元素一定得跟原数组保持一次,只要数量不相同则说明不符合转换规则,返回原数组即可。

所以最简单的思路就是将原数组进行降维,将其变为一维数组,之后执行r次裁剪操作,每次裁剪c个元素构成一个子数组,将其加入到需要返回的新数组中即可。

直接上代码:

/**
 * @param {number[][]} nums
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */
var matrixReshape = function (nums, r, c) {
    // 先判断处理之后的数组长度是否跟之前相等
    // 小知识,不管几维数组,转字符串后再还原成数组都可以做到降维的效果
    let arr = nums.join().split(',');
    if (arr.length !== r * c) {
        return nums;
    };
    // 现在要做的就是执行r次裁剪操作,每次取c个元素组成一个子数组,并加入到新数组
    const res = [];
    for(let i = 0;i < r;i++){
      	// splice会修改原数组,因此我们每次都从0开始裁剪c个是完全没问题的
        res.push(arr.splice(0,c));
    }
    return res;
};

叁 ❀ 数组降维几种方式

在做这道题之前,我理解的数组降维最直观的做法就是使用flat。flat接受一个数组并对其做降维操作,默认只降维一层,比如:

// 这是一个二维数组
let arr = [[1],[2]];
// 默认降维一次,因为会得到一个一维数组
let arr_ = arr.flat();
// 返回一个新数组,不会修改原数组
arr//[[1],[2]]
arr_//[1,2]

flat并不会修改原数组,而是在降维后返回一个新数组。falt可以接受一个参数,表示你希望降维的次数,比如我们有一个三维数组,想降成一维,传递的参数就应该是2:

[1,[2,[3]]].flat(2);//[1,2,3]

但事实上,我们可能要做降维操作,但不知道数组是几维,因此可以传递传输为Infinity,这样不管是几维,都将得到一个新的一维数组:

[[[[[1]]]]].flat(Infinity);//[1]

当然,我们可以自己模拟实现一个flat,比如这样:

function flat_(arr) {
    if (!Array.isArray(arr)) {
        throw new Error('The argument must be an array.');
    };
    return arr.reduce((accumulator, currentValue) => {
        return accumulator.concat(Array.isArray(currentValue) ? flat_(currentValue) : currentValue);
    }, []);
};

而在今天做了这道题之后,我发现原来将数组转为字符串,再还原成数组也能做到数组降维的操作,比如

([1,[2],[3,[4]]]+'').split(',')// [1,2,3,4]

而这里的原理仅仅是用了数组与字符串相加的隐式转换。同理,join也能将数组还原成字符,因此下面这段代码拥有相同效果:

[1,[2],[3,[4]].join().split(',')// [1,2,3,4]

马上11点了,发版也完成了,那么本文结束,下班下班!

posted on 2021-02-19 22:35  听风是风  阅读(246)  评论(0编辑  收藏  举报