题目:
将一个给定字符串 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"
解析:
此题目按字符串顺序按行读取即可实现。即先从第1行到第Rows行存放,然后再从Rows-1行到第2行存放,再从第1行到Rows行存放,如此往复。直到字符串取玩即可。
代码:
public String convert(String s, int numRows) {
if(numRows < 2) return s;
List<StringBuilder> rows = new ArrayList<>();
for(int i = 0; i < numRows; i++) rows.add(new StringBuilder());
//flag代表当行数在第一行或者最后一行时,行数发生相反方向的折返
int i = 0, flag = -1,lastRow = numRows -1;
for(char c : s.toCharArray()) {
rows.get(i).append(c);
if(i == 0 || i ==lastRow ){
flag = - flag;
}
i += flag;
}
StringBuilder res = new StringBuilder();
for(StringBuilder row : rows) res.append(row);
return res.toString();
}