将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：
1
2
3
| P A H N
A P L S I I G
Y I R
|
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。
请你实现这个将字符串进行指定行数变换的函数：
1
| string convert(string s, int numRows);
|
示例 1：
1
2
| 输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"
|
示例 2：
1
2
3
4
5
6
7
| 输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P I N
A L S I G
Y A H R
P I
|
示例 3：
1
2
| 输入：s = "A", numRows = 1
输出："A"
|
按顺序遍历字符串 s ：
strs[i] += c： 把每个字符 c 填入对应行
i += flag： 更新当前字符 c 对应的行索引；
flag = - flag： 在达到 Z字形转折点时，执行反向
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
| class Solution {
public String convert(String s, int numRows) {
if(numRows <= 1) return s;
StringBuilder[] strs = new StringBuilder[numRows];
for(int i = 0; i < numRows; i++){
strs[i] = new StringBuilder();
}
char[] chars = s.toCharArray();
int i = 0;
int flag = -1;
for(char c : chars){
strs[i].append(c);
if(i == 0 || i == numRows-1){
flag = -flag;
}
i += flag;
}
StringBuilder res = new StringBuilder();
for(StringBuilder sb : strs){
res.append(sb);
}
return res.toString();
}
}
|