作者:MJ昊
公众号:程序猿的编程之路
今天是 昊 的算法之路第7天,今天分享的是LeetCode第2624题蜗牛排序的解题思路。这道题目虽然难度为中等
,但通过巧妙的遍历逻辑可以轻松实现螺旋排序。
题目描述简要回顾
给定一个一维数组,你需要将其按“蜗牛形状”重新排列成一个 rowsCount x colsCount
的二维数组。如果给定数组长度和指定的行数、列数不匹配,返回一个空数组。
所谓“蜗牛形状”意味着数据以“Z”字形排列,从顶部向下,再从底部向上,依次来回摆动。
解题思路
这道题的核心思路是构建一个二维数组,并通过一个变量控制遍历的方向。我们可以依次遍历原始数组的每个元素,然后根据当前方向决定是向下填充还是向上填充,直到完成所有元素的分布。
代码实现:
Array.prototype.snail = function (rowsCount, colsCount) { if (this.length !== rowsCount * colsCount) { return []; // 如果数组长度不匹配指定的行数和列数,返回空数组 } let res = []; for (let i = 0; i < rowsCount; i++) { res.push([]);// 初始化二维数组 } let seq = true; // 控制方向的标志,true 表示向下,false 表示向上 let start = 0;// 当前要填充的行索引 for (let i = 0; i < this.length; i++) { res[start].push(this[i]);// 将元素放入对应的行 // 更新当前行的索引,根据方向决定向上或向下 if (seq) { if (start === rowsCount - 1) { seq = false;// 到达最后一行,改变方向 } else { start++;// 继续向下 } } else { if (start === 0) { seq = true;// 到达第一行,改变方向 } else { start--;// 继续向上 } } } return res; }
复杂度分析
-
时间复杂度:O(n),其中
n
是输入数组的长度。我们遍历输入数组中的每个元素一次,并将其放入正确的位置。 -
空间复杂度:O(n),需要一个大小为
rowsCount x colsCount
的二维数组来存储结果。
总结
这道题展示了如何通过合理的方向控制来遍历数组,并将其按要求转换为二维数组的形式。通过简单的方向变量控制,可以有效实现“蜗牛排序”算法。这种模式在处理二维数据排列时非常有用,尤其在矩阵遍历等问题中常见。