LeeCode 6. Z 字形变换

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

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

P      A       H      N
A  P  L  S   I   I    G
Y       I       R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。请你实现这个将字符串进行指定行数变换的函数：string convert(string s, int numRows);
示例 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"
提示：1 <= s.length <= 1000
s 由英文字母（小写和大写）、',' 和 '.' 组成
1 <= numRows <= 1000

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function(s, numRows) {
    const n = s.length, r = numRows;
    if (r === 1 || r >= n) {
        return s;
    }
    const mat = new Array(r).fill(0);
    for (let i = 0; i < r; ++i) {
        mat[i] = [];
    }
    const t = 2*r-2;
    let x = 0;
    for(let i=0;i<n;i++){
        mat[x].push(s[i]);
        if (i % t < r - 1) {
            ++x;
        } else {
            --x;
        }
    }
    let ans = [];
    for (const row of mat) {
        ans.push(row.join(''));
    }
    return ans.join('');
};
const s = "PAYPALISHIRING";
const numRows = 4;
let res = convert(s, numRows);
console.log(res)

/*
1       9           17
2     8 10       16 18
3   7   11    15    19
4 6     12 14       20
5       13          21

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