LeetCode 6. Z 字形变换(JavaScript)

148 阅读1分钟

将一个给定字符串 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"

解法:

整体的思路是遍历字符串,遍历过程中将每行都看成新的字符串构成字符串数组,最后再将该数组拼接起来即可。如果 numRows=1则说明当前字符串即为结果,直接返回 否则整个字符串需要经历,向下向上,向上向下,这样的反复循环过程。 时间复杂度:O(n),n为字符串s的长度

一句话总结:控制每个字符,应该将该字符拼接到对应的数组元素(利用变量控制字符输出的位置),最后arr.join('')输出。

    var convert = function (s, numRows) {
    if ( numRows === 1) {
                return s;
            }
            let arr = new Array(numRows).fill('');
            let model = '';
            // j来控制新字符拼接到哪一行
            let j = 0;
            debugger;
            for (let i = 0; i < s.length; i++) {
                arr[j] += s[i];
                if (j === numRows - 1) {
                    model = 'substract';
                }
                if (j === 0) {
                    model = 'add';
                }
                if (model === 'add') {
                    j++;
                } else {
                    j--;
                }
            }
            return arr.join('')
};