问题描述
6.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
题解
- 特例当行数为1时,直接返回原字符串
- 定义一个字符串的集合分别存储每行的字符,并初始化为""空字符串。
- 定义一个index作为集合的索引,用于取到每行进行拼接
- 定义一个开关loop用于控制index呈Z型来回取值
- 遍历原字符串,按顺序拼接到各行,遍历完后再将各行的字符串拼接在一起即可
class Solution {
public String convert(String s, int numRows) {
if(numRows == 1) {
return s;
}
List<StringBuilder> list = new ArrayList<StringBuilder>();
for (int i = 0; i < numRows; i++) list.add(new StringBuilder(""));
int index = 0;
boolean loop = true;
for (int i = 0; i < s.length(); i++) {
list.get(index).append(s.charAt(i));
index += loop? 1 : -1;
if (index == 0 || index == numRows - 1) loop = !loop;
}
for (int i = 1; i < list.size(); i++) list.get(0).append(list.get(i));
return list.get(0).toString();
}
}