原题地址:https://leetcode-cn.com/problems/zigzag-conversion/

一、描述

将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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"

解释:

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

示例2:

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

解释:

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

从上面的题解中可以发现规律,首先,输入字符串进行遍历,每个字符按照行先往下排列,然后到了最后一行的时候,在折返,按照行往上依次往上排列

三、JavaScript 解题

不要被示例中的空格迷惑了,真实的示例结果应该是下面这样子的:

L C I R
ETOESIIG
E D H N

再次去掉所有的空格:

LCIR
ETOESIIG
EDHN

最终拼起来得到结果字符串:LCIRETOESIIGEDHN

因此解题的时候可以按照这个思路来,从 0 行开始:

  • 0 行加一个字符 L
  • 1 行加一个字符 C
  • 2 行加一个字符 I
  • 1 行加一个字符 R
  • 0 行加一个字符 E
  • 1 行加一个字符 T
  • ....

代码如下:

/**
 * @param {string} s
 * @param {number} numRows
 * @return {string}
 */
var convert = function (s, numRows) {
    let list = new Array(); // 构造一个数组
    if (numRows === 1) return s; // 如果只有一行,不需要转换,直接输出
    let currentRow = 0; // 当前行的索引
    let goDown = false; // 是否需要变换方向(行向上还是向下)
    for (let i = 0; i < s.length; i++) {
        list[currentRow] += s[i]; // 当前行加上当前字符
        if (currentRow === 0 || currentRow === numRows - 1) {
            goDown = !goDown; // 第一行和最后一行的时候需要转换方向
        }
        currentRow += goDown ? 1 : -1; // 通过 goDown 判断是否 row 往上还是往下
    }
    let res = ''
    for (let i = 0; i < list.length; i++) {
        let tmpStr = list[i].split('undefined').join(''); // 去掉 undefined 字符
        res += tmpStr; // 拼接字符
    }
    return res;
};

四、结果

1.jpg