将一个给定字符串 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"
题解
-
思路:
- 排列方式固定:下上下上…
- 函数固定
- 设置一个字符串数组分别存储每行字符
- 循环输入每个字符存入对应行
-
第一次代码:
/**
* @param {string} s
* @param {number} numRows
* @return {string}
*/
var convert = function(s, numRows) {
let arrStr = new Array(numRows); // 根据函数设置字符串数组
for(let i = 0; i < numRows; i++) { // 初始化
arrStr[i] = '';
}
for(let i = 0; i < s.length;) { // 排列(上下为一个周期)
// 判断i不超过字符串长度,避免出现 undefined
for(let j = 0; j < numRows && i < s.length; j++) { // z 型向下的部分 依次将字符加入对应行
arrStr[j] += s[i];
i++;
}
for(let j = numRows - 2; j > 0 && i < s.length; j--) { // z 型斜向上的部分依次加入对应行
arrStr[j] += s[i];
i++;
}
}
let str = arrStr.join('');
return str;
};
- 优化
- 通过设置 flag 判断排列方向向上还是向下
/**
* @param {string} s
* @param {number} numRows
* @return {string}
*/
var convert = function(s, numRows) {
if(s.length <= numRows || numRows === 1) { // 如果字符串长度小于行数 或 行数为1,排列前后顺序不变
return s;
}
let arrStr = new Array(numRows); // 根据函数设置字符串数组
let flag = true; // 控制排列方向
let num = 0; // 字符应加入的行号
for(let i = 0; i < numRows; i++) { // 初始化
arrStr[i] = '';
}
for(let i = 0; i < s.length; i++) {
arrStr[num] += s[i];
// 改变字符应加入的行号
if(flag) { // 向下移动
num++;
}
else { // 向上移动
num--;
}
// 判断移动方向
if(num == numRows - 1) { // 向下移动
flag = false;
}
else if (num == 0) { // 向上移动
flag = true;
}
}
let str = arrStr.join('');
return str;
};