题目描述:
请你编写一段代码为所有数组实现 snail(rowsCount,colsCount)
方法,该方法将 1D 数组转换为以蜗牛排序的模式的 2D 数组。无效的输入值应该输出一个空数组。当 rowsCount * colsCount !==
nums.length
时。这个输入被认为是无效的。
蜗牛排序从左上角的单元格开始,从当前数组的第一个值开始。然后,它从上到下遍历第一列,接着移动到右边的下一列,并从下到上遍历它。将这种模式持续下去,每列交替变换遍历方向,直到覆盖整个数组。例如,当给定输入数组 [19, 10, 3, 7, 9, 8, 5, 2, 1, 17, 16, 14, 12, 18, 6, 13, 11, 20, 4, 15]
,当 rowsCount = 5
且 colsCount = 4
时,需要输出矩阵如下图所示。注意,矩阵沿箭头方向对应于原数组中数字的顺序
题目分析:
首先题目说明当rowsCount * colsCount !== nums.length时。这个输入被认为是无效的。所以我们可以直接判断当rowsCount * colsCount !== this.length时,返回一个空数组即可。我们通过观察原数组的排列发现,可以把原数组看成是一个s型的排列顺序,想要输出题目所需矩阵,首先,我们先将原数组的每一列都整理成从上到下的顺序,然后再进行赋值,变为每一行从左到右的顺序,这里需要开辟一个答案数组。
源代码:
主要通过两个双重循环来完成,第一个整理原数组的排列顺序,第二个对答案数组进行赋值。
/**
* @param {number} rowsCount
* @param {number} colsCount
* @return {Array<Array<number>>}
*/
Array.prototype.snail = function(rowsCount, colsCount) {
if(rowsCount * colsCount !== this.length) {
return new Array();
}
const ansArr = new Array(rowsCount).fill(0).map((item) => {return item = new Array(colsCount).fill(0);});
// 先把数组整理成每一列从上到下的顺序
for(let i = rowsCount; i < this.length; i += rowsCount * 2) {
let l = i, r = i + rowsCount - 1;
while(l < r) {
[this[l], this[r]] = [this[r], this[l]];
++l, --r;
}
}
// 开始赋值,数组变为每一行从左到右的顺序
for(let i = 0; i < this.length; i += rowsCount) {
for(let j = i; j < i + rowsCount; ++j) {
ansArr[j - i][i / rowsCount] = this[j];
}
}
return ansArr;
}
/**
* const arr = [1,2,3,4];
* arr.snail(1,4); // [[1,2,3,4]]
*/