将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
解题思路:
使用StringBuilder[]数组来存储z字形变换的结果,以实例2为例,如下图所示:
可以将Z字形过程分为两个部分,一个是从[0, numRows-1]向下依次存储,然后从[numRows-2, 1]向上依次存储,后面就是这两部分的循环即可。
最后将所有StringBuilder集合到sbs[0]中输出即可。
public String convert(String s, int numRows) {
if(numRows <= 1)
return s;
char[] ss = s.toCharArray();
StringBuilder[] sbs = new StringBuilder[numRows];
for(int i=0; i<numRows; i++)
sbs[i] = new StringBuilder();
int index = 0;
while(index < ss.length){
//向下
for(int i=0; i<numRows && index < ss.length; i++)
sbs[i].append(ss[index++]);
//向上
for(int i=numRows-2; i>0 && index < ss.length; i--)
sbs[i].append(ss[index++]);
}
for(int i=1; i<numRows; i++)
sbs[0].append(sbs[i]);
return sbs[0].toString();
}