[算法基础] 字符串操作-Z字形变换

题目

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

比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N

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

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

string convert(string s, int numRows);

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

解题思路

对于这类问题，我们需要寻找规律。而不是用固定算法求解。

从图中可以看出，一个V型对应这一个循环，那么一个循环内，元素个数为多少呢？n = ( 2 * numRows -2 )，是2倍的行数减去头尾两个没有占用的字符。对于原字符索引i，可以计算r = i % n。
在重新排列的字符串中，行号r有如下规律：

• 当r < numRows 时，r为对应行号。
• 当r >= numRows 时，n - r 为对应行号。

行
0	[0] % 6 = [0]			[6] % 6 = [0]
1	[1] % 6 = [1]		[5] % 6 = 5; 6-5 = [1]	[7] % 6 = [1]
2	[2] % 6 = [2]	[4] % 6 = 4; 6-4 = [2]		[8] % 6 = [2]
3	[3] % 6 = [3]

根据规律可以拼接出每一行的字符串，最后合并所有行即可。

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
convert = function (s, numRows) {
    if (numRows !== 1) {
    	//用空字符串填充数组，默认是undefined.也为后续字符串相连做准备。
        let arr = new Array(numRows).fill("");
        
        let n = numRows * 2 - 2;//循环周期
        for (let i = 0; i < s.length; i++) {
            let r = i % n;
            // 在 arr 数组中，拼接各行的字符串
            arr[r < numRows ? r : n - r] += s[i];//V字形走位规律
        }
        return arr.join("");// 把各行的字符串拼接并返回
    } else {
        return s;
    }
};