题目
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
思路
一开始思路被限制了,想着通过坐标计算得到最终结果,无奈规律实在太难找了,后面又想着用二维数组解题,但是想着又发现空间复杂度太大了,后来看了下官方的解题思路,才恍然大悟,之前的思路一直围绕在一次性求出结果字符串,实际上先按行或者按列求出每行每列,再拼接,这道题就会特别简单,我这里采用先求每行再拼接的思路,代码如下:
代码
public String convert(String s, int numRows) {
if (numRows == 1)
return s;
List<StringBuilder> temp = new ArrayList<>();
for (int i=0; i<Math.min(numRows, s.length()); i++)
temp.add(new StringBuilder());
int curRow = 0, direction = 1;
for (char c : s.toCharArray()){
temp.get(curRow).append(c);
if (curRow == 0)
direction = 1;
if (curRow == numRows-1)
direction = -1;
curRow += direction;
}
StringBuilder res = new StringBuilder();
temp.forEach(res::append);
return res.toString();
}