Z字形变换 Java
题目描述:
将一个给定字符串 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"
2、
输入:s = "PAYPALISHIRING", numRows = 4
输出:"PINALSIGYAHRPI"
解释:
P I N
A L S I G
Y A H R
P I
3、
输入:s = "A", numRows = 1
输出:"A"
解答:
class Solution {
public String convert(String s, int numRows) {
//这种特殊情况下的输入,直接返回字符串即可
if(numRows==1||s.length()<=numRows){
return s;
}
char[] c = s.toCharArray();
//创建一个StringBuilder数组,每个StringBuilder分别存储一行
StringBuilder [] ss=new StringBuilder[numRows];
//给每个StringBuilder初始化
for (int i = 0; i < numRows; i++) {
ss[i]=new StringBuilder();
}
int now_row=1;//确定当前字符应处于哪一行
//当upOrDown=true,字符所处的行数应递增,否则递减
boolean upOrDown=true;
// re用于存储最终拼接结果
StringBuilder re=new StringBuilder();
//遍历每一个字符,并根据他们所处行append加入到不同的StringBuilder对象中
for (int i=0;i<c.length;i++){
ss[now_row-1].append(c[i]);
if (i!=0&&(now_row==numRows||now_row==1)){
upOrDown=!upOrDown;
}
if (upOrDown==true){
now_row++;
}else {
now_row--;
}
}
for (int i = 0; i < numRows; i++) {
re.append(ss[i]);
}
return re.toString();
}
}
另外求代码优化建议!!!