描述
将一个给定字符串 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);
分析
建立一个StringBuilder的数组,遍历字符串的同时也遍历StringBuilder数组,把s的字符添加在StringBuilder里,StringBuilder数组的下标是上下来回变化,这样刚好实现题目Z型上下的效果,又因为StringBuilder.append,使得字符是向右累放置的。
class Solution {
public String convert(String s, int numRows) {
if(numRows == 1){
return s;
}
StringBuilder[] sb = new StringBuilder[numRows];
for(int i = 0; i < numRows; i++){
sb[i] = new StringBuilder();
}
int row = 0;
int index = 0;
int n = s.length();
while(index < n){
while(index < n && row < numRows){
char ch = s.charAt(index++);
sb[row++].append(ch);
}
row = numRows - 2;
while(index < n && row >= 0){
char ch = s.charAt(index++);
sb[row--].append(ch);
}
row = 1;
}
StringBuilder stb = new StringBuilder();
for(StringBuilder str : sb){
stb.append(str);
}
return stb.toString();
}
}