L 字形变换
题目描述
将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 L 字形排列。
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串.
请你实现这个将字符串进行指定行数变换的函数：
convert(s:string, numRows:number ) => string
比如输入字符串为 "ABCDEFGHIJKLMNO" 行数为 3 时，排列如下：
(-表示占位符)
A--F--K
B--G--L
CDEHIJMNO
输出：AFKBGLCDEHIJMNO
行数为 4 时，排列如下：
A---H---O
B---I---
C---J---
DEFGKLMN
输出：AHOBICJDEFGKLMN
行数为 5 时，排列如下：
A----J
B----K
C----L
D----M
EFGHINO
输出：AJBKCLDMEFGHINO
方法一：利用二维矩阵模拟
/**
* 方法一：利用二维矩阵模拟
* @param {string} s
* @param {number} numRows
* @return {string}
*/
const convert_a = (s, numRows) => {
const n = s.length, r = numRows;
if (r == 1 || r >= n) {
// r==1 只有一行，r>=n 只有一列。
return s;
}
const t = r * 2 - 1; // L 字形变换的周期 t = r + r - 1
// 每个周期会占用矩阵上的 1+r-1 列，即 r 列。
// 因此我们有 Math.ceil(n/t) 个周期（最后一个周期视作完整周期），乘上每个周期的列数，得到矩阵的列数 c = Math.ceil(n/t) * r。
const c = Math.ceil(n / t) * r; // c 为矩阵的列数，最后一个周期视作完整周期。
const c_ = Math.floor((n + t - 1) / t) * r; // c 的其他运算方法。
// 创建一个 r 行 c 列的矩阵。
const matrix = new Array(r).fill(0).map(() => new Array(c).fill(0));
// 遍历字符串 s 并按 L 字形填入。
// 设当前填写的位置为 (x,y)，即矩阵的 x 行 y 列。初始 (x,y)=(0,0)，即矩阵左上角。
// 若当前字符下标 i 满足 i mod t < r - 1，则向下移动，否则向右移动。
// 若当前字符下标 i 满足 (i+1) mod t == 0，表示将开始新的周期，向上移动回第一行。
for (let i = 0, x = 0, y = 0; i < n; ++i) {
matrix[x][y] = s[i];
if (i % t < r - 1) {
++x; // 向下移动
} else {
++y; // 向右移动
}
if ((i + 1) < n && (i + 1) % t == 0) {
x = 0; // 回到新周期第一个字符。
}
}
// 填写完成，逐行扫描矩阵中的非空字符，组成答案。
const ans = [];
for (const row of matrix) {
for (const char of row) {
if (char !== 0) {
ans.push(char);
}
}
}
return ans.join('');
};
方法二：压缩矩阵空间
/**
* 方法二：压缩矩阵空间
* @param {string} s
* @param {number} numRows
* @return {string}
*/
const convert_b = (s, numRows) => {
// 方法一中的矩阵有大量的空间没有被使用，优化。
// 注意到每次往矩阵的某一行添加字符时，都会添加到该行上一个字符的右侧，且最后组成答案时只会用到每行的非空字符。
// 因此我们可以将矩阵的每行初始化为一个空列表，每次向某一行添加字符时，添加到该行的列表末尾即可。
const n = s.length, r = numRows;
if (r == 1 || r >= n) {
// r==1 只有一行，r>=n 只有一列。
return s;
}
const matrix = new Array(r).fill(0).map(() => []);
for (let i = 0, x = 0, t = r * 2 - 1; i < n; ++i) { // L 字形变换的周期 t = r + r - 1
matrix[x].push(s[i]);
if (i % t < r - 1) {
++x;
}
if ((i + 1) < n && (i + 1) % t == 0) {
x = 0; // 回到新周期第一个字符。
}
}
const ans = [];
for (const row of matrix) {
ans.push(row.join(''));
}
return ans.join('');
};
方法三：直接构造
/**
* 方法三：直接构造
* @param {string} s
* @param {number} numRows
* @return {string}
*/
const convert_c = (s, numRows) => {
// 研究下方法一中矩阵的每个非空字符会对应到 s 的哪个下标（记作 idx），从而直接构造出答案。
// 由于 L 字形变换的周期为 t = r * 2 - 1，
// 因此对于矩阵第一行的非空字符，其对应的 idx 均为 t 的倍数，即 idx (mod t) ≡ 0。
// 同理，对于矩阵第二行、第三行、非最后一行（行号设为 i）的非空字符，应满足 idx (mod t) ≡ i。
// 对于矩阵的最后一行（行号为 r - 1），每个周期内有四个字符，
// 第一个字符满足 idx (mod t) ≡ i ≡ r - 1，
// 第二个字符满足 idx (mod t) ≡ t - i ≡ t - r + 1，
// 第三个字符满足 idx (mod t) ≡ t - i + 1 ≡ t - r + 2，
// 第四个字符满足 idx (mod t) ≡ t - i + 2 ≡ t - r + 3。
const n = s.length, r = numRows;
if (r == 1 || r >= n) {
// r==1 只有一行，r>=n 只有一列。
return s;
}
const t = r * 2 - 1; // L 字形变换的周期 t = r + r - 1
const ans = [];
for (let i = 0; i < r; i++) { // 枚举矩阵的行
for (let j = 0; j < n - i; j += t) { // 枚举每个周期于该行的起始下标 // 判断起始下标没有超出字符串长度，即 j + i < n，即 j < n - i。
ans.push(s[j + i]); // 非最后一行，及最后一行的周期内最前的下标位置。
if (i == r - 1) {
for (let k = r - 1; k > 0 && (j + t - k) < n; --k) { // 判断下标没有超出字符串长度，即 j + t - k < n
ans.push(s[j + t - k]);
}
}
}
}
return ans.join('');
};
运行结果
