记录个人算法学习6：Z 字形变换

问题描述

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

示例 1：

输入： s = "PAYPALISHIRING", numRows = 3
输出： "PAHNAPLSIIGYIR"

示例 2：

输入： s = "PAYPALISHIRING", numRows = 4
输出： "PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：

输入： s = "A", numRows = 1
输出： "A"

解题思路

这道题我第一感觉也不难，定义numRows个数组，然后挨个判断应该在哪个数组里面，最后合并下数组，但是难点在于普通人判断应该在哪个数组里面时可能会想到找规律的方法，也就是说判断索引和行数的关系，但是那样是行不通的，这时候就要尽量的用算法指针的一个概念，根据指针所在行数去判断应该放在哪个数组里面

1. 时间复杂度要控制在O(n)
2. 在for循环中判断每个字符应该在哪个位置
3. 巧用指针

代码实现

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function(s, numRows) {
    if (numRows < 2) return s;
    let res = new Array(numRows).fill("");
    let i = 0, flag = -1;

    for (let c of s) {
        res[i] += c;
        if (i === 0 || i === numRows - 1) flag = -flag;
        i += flag;
    }

    return res.join('');
};