題意
將一個(gè)給定字符串 s 根據(jù)給定的行數(shù) numRows 壳鹤，以從上往下京髓、從左到右進(jìn)行 Z 字形排列盟劫。
比如輸入字符串為 "PAYPALISHIRING" 行數(shù)為 3 時(shí)夜牡，排列如下：
P A H N
A P L S I I G
Y I R
之后，你的輸出需要從左往右逐行讀取侣签，產(chǎn)生出一個(gè)新的字符串塘装，比如："PAHNAPLSIIGYIR"。
請(qǐng)你實(shí)現(xiàn)這個(gè)將字符串進(jìn)行指定行數(shù)變換的函數(shù)：
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"
提示：
1 <= s.length <= 1000
s 由英文字母（小寫和大寫）影所、',' 和 '.' 組成
1 <= numRows <= 1000
題解
使用一個(gè)數(shù)組保存每行的字符蹦肴，行位置：根據(jù)down的方向，控制行位置猴娩，再將對(duì)應(yīng)字符存儲(chǔ)到對(duì)應(yīng)行數(shù)組中阴幌。
class Solution {
public String convert(String s, int numRows) {
if (s == null || s.length() <= 1 || numRows == 1) {
return s;
}
StringBuilder[] ress = new StringBuilder[numRows];
for (int i = 0; i < numRows; i++) {
ress[i] = new StringBuilder();
}
int curRow = 0;
boolean down = false;
for (int i = 0; i < s.length(); i++) {
// 第一行或者最后一行，控制行方向反轉(zhuǎn)
if (curRow == 0 || curRow == (numRows - 1)) {
down = !down;
}
ress[curRow].append(s.substring(i, i+1));
curRow = curRow + (down ? 1 : -1);
}
StringBuilder res = new StringBuilder();
for (int i = 0; i < numRows; i++) {
res.append(ress[i]);
}
return res.toString();
}
}
