将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时,排列如下:
L C I R
E T O E S I I G
E D H N
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"LCIRETOESIIGEDHN"。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L D R
E O E I I
E C I H N
T S G
思路:flag方向标记,list实现二维结构
class Solution {
public String convert(String s, int numRows) {
//很愚蠢的办法
//设置标记
int index = 0;
//设置方向
boolean flag = true;
//如果numRows>s.length()
int len = s.length();
if(numRows>=len||numRows==1){
return s;
}
//结果集
List<StringBuffer> list = new ArrayList<>();
//填入创建结果行
for(int i = 0;i<numRows;i++){
StringBuffer sb = new StringBuffer();
list.add(sb);
}
//遍历字符串
for(int i=0;i<len;i++){
//当前存储方向(flag==true,向下,flag==false向上)
if(index>=numRows){
flag = false;
index=index-2;
}
if(index<0){
flag = true;
index=index+2;
}
//存入当前字符
list.get(index).append(s.charAt(i));
//调整下标
if(flag){
index++;
}else{
index--;
}
}
StringBuffer result = new StringBuffer();
//组合结果
for(StringBuffer sb : list){
result.append(sb.toString());
}
return result.toString();
}
}