LeetCode|2624. 蜗牛排序(day7)

作者：MJ昊

博客：掘金、CSDN等

公众号：程序猿的编程之路

今天是 昊 的算法之路第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 的二维数组来存储结果。

总结

这道题展示了如何通过合理的方向控制来遍历数组，并将其按要求转换为二维数组的形式。通过简单的方向变量控制，可以有效实现“蜗牛排序”算法。这种模式在处理二维数据排列时非常有用，尤其在矩阵遍历等问题中常见。