算法题-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

解答

var convert = function(s, numRows) {
    let arr = []
    let j = 0
    for(let i = 0; i < numRows; i++){
        arr[i] = ''
    }
    for(let n of s){
        j = j >= 2 * (numRows - 1) ? (j - 2 * (numRows - 1)) : j
        let k = j > (numRows - 1) ? 2 * (numRows - 1) - j : j
        arr[k] += n
        j++
    }
    return numRows === 1 ? s : arr.join('')
};

分析

时间复杂度：O(n)，空间复杂度：O(n)