题目描述

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

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

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

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

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

conver(s, numRows)

示例一🌰

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

示例二🌰

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

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

解题思路

1. 考虑字符串长度为0或者1的情况;
2. 考虑输入的numRows为0或者小于字符串长度的情况;
3. 通过比较字符串的长度和numRows来定义一个行的最小长度len;
4. 定义当前的下标loc和是否向下移动down;
5. 判断loc是否是在列的头或者列的尾部,从而改变移动的方向down;
6. down决定了loc是+1还是-1

coding

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function(s, numRows) {
    if (!s || !numRows) return ''
    let len = Math.min(s.length, numRows)
    if (len === 1) return s
    
    let array = Array.from({length: len}, i => '')
    let loc = 0,
        down = false;
    let rstStr = ''
    for (const c of s) {
        array[loc] += c;
        if (loc === 0 || loc === numRows - 1) // 在列的头部或者尾部
            down = !down
        loc += down ? 1 : -1
    }
    return array.join('');
};

测试代码

convert('A', 2)
// 'A'
convert('AB', 1)
// 'AB'
convert('LinDaiDai', 3)
// 'LaiiDianD'

阅读全文