题目描述

将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时,排列如下:

P   A   H   N
A P L S I I G
Y I R

之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"

示例

示例1.

输入:s = "PAYPALISHIRING", numRows = 3
输出:"PAHNAPLSIIGYIR"
输入:s = "PAYPALISHIRING", numRows = 4
输出:"PINALSIGYAHRPI"
解释:
P I N
A L S I G
Y A H R
P I

题解

  1. 通过观察可以发现,字符的索引排列位置是按照行的方向从小到大,再从大到小排列的
  2. 可以按行数给每层创建一个StringBuilder保存每行存储的字符串
  3. 维护一个变量表示索引是递增还是递减,遇到最后一行就递减,遇到第一行就递增

Coding

class Solution {
public String convert(String s, int numRows) {
if(numRows < 2){
return s;
}
List<StringBuilder> res = new ArrayList<>();
for(int i = 0;i < numRows;i++){
res.add(new StringBuilder());
}
int i = 0;
int flag = -1;
for(char c:s.toCharArray()){
res.get(i).append(c);
if(i == 0 || i == numRows - 1){
flag = -flag;
}
i += flag;
}
StringBuilder sb = new StringBuilder();
for(StringBuilder ss:res){
sb.append(ss);
}
return sb.toString();
}
}