Z字型变换
将一个给定字符串 s
根据给定的行数 numRows
,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING"
行数为 3
时,排列如下:
P A H N
A P L S I I G
Y I R
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"
。
- 思路:
假设行数为 3 ,故设置 3 个 List(res[0],res[1],res[2]),将每次遍历的值分别放入 0 1 2 1 0 1 2 依次类推。
P ----> res[0] A ----> res[0]
A ----> res[1] P ----> res[1] L ----> res[1]
Y ----> res[2] I ----> res[2]
这样在 res[0] 中存放的都是 第一行的数据;将res[0],res[1],res[2] 依次打印,就得到结果;
class Solution {
public String convert(String s, int numRows) {
if(numRows < 2){
return s;
}
List<StringBuilder> res = new ArrayList<StringBulider>();
//为list赋值
for(int i = 0;i < numRows;i++){
res.add(new StringBuilder());
}
//利用flag 来设置 i 的值,让其每次都跳过 0 或 3
int i = 0 ,flag = -1;
// i:0 c:P flag:1
// i:1 c:A flag:1
// i:2 c:Y flag:-1
// i:1 c:P flag:-1
// i:0 c:A flag:1
for(char c : s.toCharArray()){
res.get(i).append(c);
if(i == 0 || i = numRows -1){
flag = -flag;
}
i += flag;
}
StringBuilder result = new StringBuilder();
for(StringBuilder result : res){
result.append(row);
}
return result.toString();
}
}